原题链接:https://leetcode-cn.com/problems/maximum-average-subarray-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]。
思路
遍历计算平均值输出其中最大的。
这样子可以做,但是超时。那换个思路。
因为是连续子数组,所以对于给定的数组的顺序应该是不能去做改变的。
那求和的时候就不都求了,就减去第一个然后加上最后一个这样子吧,减少运算量。
傻逼中的傻逼,讲个笑话,DBL_MIN是2.22507e-308,INT_MIN是-2147483648。
class Solution {
public:
double findMaxAverage(vector<int>& nums, int k) {
double sum = 0;
//求出第一组数据和
for(int i =0; i < k; i++) {
sum += nums[i];
}
double tmp = sum;
for(int i = 0; i < nums.size() - k; i++) {
//减去第一个数加上后一个数,比较大小后留下较大的和
tmp = tmp - nums[i] + nums[i + k];
if(tmp > sum) sum = tmp;
}
return sum/k;
}
};
题解中有个思路叫preSum,字面意思就是前置的和。
该思想是这样子的,假设有个数组,有n个数,我们新设一个数组,就叫preSum,有n+1个数,preSum[i]呢就表示数组前i个数的和,不包括i。而这个preSum数组只需要一遍遍历就可以得到。有这个preSum数组呢就可以求任意区间的长度了,在该题中长度是k,那么就只要再遍历一遍找到k区间长度的最大值就行了。