LeetCode 每日一題643. 子陣列最大平均數 I
阿新 • • 發佈:2021-02-11
643. 子陣列最大平均數 I
給定 n 個整數,找出平均數最大且長度為 k 的連續子陣列,並輸出該最大平均數。
示例:
輸入:[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]。
方法一:
- 本題比較簡單,可以做為「滑動視窗」和「字首和」的入門題。
參考程式碼 1:滑動視窗
public double findMaxAverage(int[] nums, int k) {
double sum = 0;
for(int i = 0; i < k; i++) {
sum += nums[i];
}
double ret = sum;
for (int i = k; i < nums.length; i++) {
sum = sum - nums[i - k] + nums[i];
ret = Math.max(ret, sum);
}
return ret / k;
}
參考程式碼 2:字首和
public double findMaxAverage (int[] nums, int k) {
int n = nums.length;
for (int i = 1; i < n; i++) {
nums[i] = nums[i - 1] + nums[i];
}
double ret = nums[k - 1];
for (int i = k; i < n; i++) {
ret = Math.max(ret, nums[i] - nums[i - k]);
}
return ret / k;
}
執行結果