1490: 连续子串和 （贪心）

1490: 连续子串和

http://www.acmore.net/problem.php?id=1490

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 209  Solved: 36

 

Description

小Y前段时间刚刚做完连续子串和问题，相信大家对连续子串和也都不陌生，现在小Y又碰到了一个新的难题，给定N个整数，求至少K个数连续的最大和为多少？

Input

若干组测试数据，每组数据第一行给出两个数N（1<=N<=10^6），K（1<=K<=N）。接下来一行N个数字，分别是a1,a2...aN，对于每个数满足-1000<=ai<=1000。

Output

输出长度大于等于K的连续整数和值中最大的一个。

Sample Input

5 3 -2 3 6 0 1

Sample Output

10

 分析：

典型贪心，要求最少k个连续整数和的最大值，从第k个数开始枚举每一个数字作为结束点。保留前i位数的前缀和为sum[i],对于以i为结束的序列，其值为

sum[i]-sum[i-k], sum[i]-sum[i-k-1], sum[i]-sum[i-k-2], sum[i]-sum[1], sum[0]之一，那么我们只需要对每一个i保留sum[0]到sum[i-k]的最小值就可以了

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;

const int maxn = 1e6+10;
const int INF = 0x3f3f3f3f;

int n,k;
int num[maxn];

int solve(){
    int tmp=INF;
    int ans=-INF;
    for(int i=k; i<=n; i++ ){
        tmp=min(tmp,num[i-k]);
        ans=max(ans,num[i]-tmp);
       /* cout<<"tmp="<<tmp<<" "<<"ans="<<ans<<endl;*/
    }
    return ans;
}

int main(){
   /* freopen("C:\Users\Brave_Wtz\Desktop\in.txt","r",stdin);
    freopen("C:\Users\Brave_Wtz\Desktop\put.txt","w",stdout);*/
    while(~scanf("%d%d",&n,&k)){
        memset(num,0,sizeof(num));
        for( int i=1; i<=n; i++ ){
            cin>>num[i];
            num[i]+=num[i-1];
        }
        cout<<solve()<<endl;
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}