FZU Problem 2168 防守阵地 I

Problem 2168 防守阵地 I

Accept: 347    Submit: 1191
Time Limit: 3000 mSec    Memory Limit : 32768 KB

Problem Description

部队中共有N个士兵，每个士兵有各自的能力指数Xi，在一次演练中，指挥部确定了M个需要防守的地点，按重要程度从低到高排序，依次以数字1到M标注每个地点的重要程度，指挥部将选择M个士兵依次进入指定地点进行防守任务，能力指数为X的士兵防守重要程度为Y的地点将得到X*Y的参考指数。现在士兵们排成一排，请你选择出连续的M个士兵依次参加防守,使得总的参考指数值最大。

Input

输入包含多组数据。

输入第一行有两个整数N,M(1<=N<=1000000，1<=M<=1000),第二行N个整数表示每个士兵对应的能力指数Xi（1<=Xi<=1000）。

对于30%的数据1<=M<=N<=1000。

Output

输出一个整数，为最大的参考指数总和。

Sample Input

5 32 1 3 1 4

Sample Output

17

这题举个例子就懂了。

n=5m=3

a[i]:   1  2  3   4   5

前三个之和：    sum  =  a[1]+a[2]+a[3]

以第一个开头：sum1  =  a[1]*1+a[2]*2+a[3]*3

以第二个开头：sum2  =  a[2]*1+a[3]*2+a[4]*3  =  sum1 - sum+a[4]*3

之后  sum更新：   sum = a[2]+a[3]+a[4] = sum - a[1] + a[4];

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define N 1000110using namespace std;int a[N];int n,m;int main() {    while(~scanf("%d%d",&n,&m)) {        for(int i=1; i<=n; i++)            scanf("%d",&a[i]);        int Max=0,sum=0;        for(int i=1; i<=m; i++) {            Max+=i*a[i];             sum+=a[i];  ///前m个数的和        }        int first=1;        int s=Max;        for(int i=m+1; i<=n; i++) {            s=s-sum+a[i]*m;                    sum=sum-a[first++]+a[i];  ///减去头一个，加上最后一个            Max=max(Max,s);        }        printf("%d\n",Max);    }    return 0;}