[LeetCode] Maximum Average Subarray I 子陣列的最大平均值
阿新 • • 發佈:2018-12-27
Given an array consisting of n
integers, find the contiguous subarray of given length k
that has the maximum average value. And you need to output the maximum average value.
Example 1:
Input: [1,12,-5,-6,50,3], k = 4 Output: 12.75 Explanation: Maximum average is (12-5-6+50)/4 = 51/4 = 12.75
Note:
- 1 <=
k
<=n
<= 30,000. - Elements of the given array will be in the range [-10,000, 10,000].
這道題給了我們一個數組nums,還有一個數字k,讓我們找長度為k且平均值最大的子陣列。由於子陣列必須是連續的,所以我們不能給陣列排序。那麼怎麼辦呢,在博主印象中,計運算元陣列之和的常用方法應該是建立累加陣列,然後我們可以快速計算出任意一個長度為k的子陣列,用來更新結果res,從而得到最大的那個,參見程式碼如下:
解法一:
class Solution { public: doublefindMaxAverage(vector<int>& nums, int k) { int n = nums.size(); vector<int> sums = nums; for (int i = 1; i < n; ++i) { sums[i] = sums[i - 1] + nums[i]; } double mx = sums[k - 1]; for (int i = k; i < n; ++i) { mx= max(mx, (double)sums[i] - sums[i - k]); } return mx / k; } };
由於這道題子陣列的長度k是確定的,所以我們其實沒有必要建立整個累加陣列,而是先算出前k個數字的和,然後就像維護一個滑動視窗一樣,將視窗向右移動一位,即加上一個右邊的數字,減去一個左邊的數字,就等同於加上右邊數字減去左邊數字的差值,然後每次更新結果res即可,參見程式碼如下:
解法二:
class Solution { public: double findMaxAverage(vector<int>& nums, int k) { double sum = accumulate(nums.begin(), nums.begin() + k, 0), res = sum; for (int i = k; i < nums.size(); ++i) { sum += nums[i] - nums[i - k]; res = max(res, sum); } return res / k; } };
參考資料: