1. 程式人生 > >131、子陣列最大平均數

131、子陣列最大平均數

題目描述
給定 n 個整數,找出平均數最大且長度為 k 的連續子陣列,並輸出該最大平均數。

示例 1:

輸入: [1,12,-5,-6,50,3], k = 4
輸出: 12.75
解釋: 最大平均數 (12-5-6+50)/4 = 51/4 = 12.75

注意:

1 <= k <= n <= 30,000。
所給資料範圍 [-10,000,10,000]。

class Solution {
    public double findMaxAverage(int[] nums, int k) {
       
        if(nums.length == 0){
            return 0.0;
        }
        if(nums.length== 1){
            return nums[0]/k;
        }
         double average_num = 0;
        for(int j=0;j<k;j++){
            average_num =average_num+  nums[j];
        }
        double tmp_ave = average_num;
        for(int i=1;i<=nums.length-k;i++){
            double tmpAver = 0;
            tmpAver = tmp_ave - nums[i-1] + nums[i+k-1];
            tmp_ave = tmpAver;
            if (average_num <= tmp_ave){
                average_num = tmp_ave;
            }
        }
        return average_num/k;
    
    }
}

寫於南京市湯山溫泉小屋,這個程式碼的思路很簡單,只是在求解總和時有點技巧

排名靠前的程式碼
思路類似

class Solution {
    public double findMaxAverage(int[] nums, int k) {
        int n=nums.length;
        int sum=0;
        int sum1=0;
        for(int i=0;i<k;i++) {
            sum1+=nums[i];
        }
        sum=sum1;
        for(int i=k;i<n;i++){
            sum1 += nums[i];
            sum1 -= nums[i-k];
            sum =Math.max(sum1,sum);
        }
        return sum*1.0/k;
    }
}