给定 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]。
思想
由于k值给定,因此只需要求出数组中连续k个和最大的数。
首先将前k个数的和保存到tmp中,res始终保存当前最大和。
将数组后面的数一个一个加到tmp中,相应的也要减去tmp中保存的第一个数。
比较tmp和res的大小,及时更新res。
1 class Solution { 2 public: 3 double findMaxAverage(vector<int>& nums, int k) { 4 double res=0; 5 int tmp=0; 6 for(int i=0;i<nums.size();i++){ 7 if(i<k){ 8 tmp+=nums[i]; 9 res=tmp; 10 continue; 11 } 12 tmp=tmp+nums[i]-nums[i-k]; 13 if(tmp>res) res=tmp; 14 } 15 return res/k; 16 } 17 };