Description | ||
题目描述给你一个数列a1,a2,...,an,求m个连续数字组成的子段和最大值。 输入有多个样例,每个样例的第一行是两个整数n和m,(1≤m≤n;≤100,000)。如果n和m为0表示输入结束,这个样例不需要处理。第二行是n个整数ai,0≤ai≤10000。 输出每行输出一个整数,即样例的结果。 样例输入6 3 1 2 3 4 5 6 6 3 1 2 3 3 2 1 0 0 样例输出15 8 | ||
Sample Input | ||
Sample Output |
数据结构有讲过一个最大子集,思路差不多,感兴趣可以去拓展一些。
这题思路其实很简单也很好理解,相信可以相出很多种解法,但是在xtu的oj上,这道题的时间卡德非常紧,为了时时间复杂度降低,我的应该是O(^n),用了这种边输入边计算的方法。
关键就是当sum加到m个之后,sum-=a[i-m],这样sum就能一直保持在只有m个数相加,并且还是连续的,最后就用max来比较每个sum,最终输出最大的max,这样是不会漏掉任何一个sum的可能结果的,因为sum的变化始终是线性连续的。
为了便于理解,模拟一下:当输入第m+1个的时候,sum减掉a[0],那么此时sum就是1~m的和,依然是m个子段的和。
#include<stdio.h>
#include<string.h>
const int MAXNUM=1e6+5;
int a[MAXNUM]={0};
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF){
if(m==0&&n==0) break;
memset(a,0,sizeof(a));// 要记得初始化为0
int max=0;
int sum=0;
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
sum+=a[i];// 每个循环都+一次a[i]
if(i>=m) sum-=a[i-m];当加到第m+1次的时候,开始减掉前面的第i-m个
if(sum>=max) max=sum;// 比较并输出
}
printf("%d\n",max);
}
return 0;
}