Spurs

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

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. 1 <= k <= n <= 30,000.
  2. Elements of the given array will be in the range [-10,000, 10,000].

自己方法1 Brute Force:
窗口k = 4, 从数组左到右滑动, 代码省略.
\(O(n*k)\) time, \(O(1)\) extra space.

自个方法2 改进版的 Brute Force:
想法如下:

k = 4 的情形

1	12	-5	-6	50	3
^			 ^

1	12	-5	-6	50	3
	^			^
明显的, 第一行的 sum  + 50  -  1 就是第二行的 sum
					A[i]  A[i-k]

\(O(n)\) time, \(O(1)\) extra space.

double findMaxAverage(vector<int>& A, int k) {
    int i, sum = 0, max = INT_MIN;
    for (i = 0; i < k; i++) sum += A[i];
    max = max > sum ? max : sum;
    for (i = k; i < A.size(); i++) {
        sum = sum + A[i] - A[i - k];
        max = max > sum ? max : sum;
    }
    return 1.0 * max / k;
}
posted on 2017-08-15 16:27  英雄与侠义的化身  阅读(84)  评论(0编辑  收藏  举报