福州大学 Problem 2168 防守阵地 I

http://acm.fzu.edu.cn/problem.php?pid=2168

最重要的是 dp[k]=dp[k-1]-ans[k-1]+x[i]*m;

ans[k-1]是m个数求和。

 Problem 2168 防守阵地 I

Accept: 14    Submit: 20 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 3

2 1 3 1 4

Sample Output

17

#include<stdio.h>
int dp[1000005],x[1000005],ans[1000005];
 int main()
{
    int n,m,k,max;
      int  i,j;
    while(~scanf("%d%d",&n,&m))
    {
             k=0;
             ans[0]=0;dp[0]=0;
          for(i=1;i<=n;i++)
          {
               scanf("%d",&x[i]);
               if(i<=m)
                 {
                    ans[k]+=x[i];
                    dp[k]+=i*x[i];
                    max=dp[k];
                 }
                else
                  {
                      k++;
                      ans[k]=ans[k-1]-x[i-m]+x[i];
                      dp[k]=dp[k-1]-ans[k-1]+x[i]*m;
                     if(max<dp[k])
                         max=dp[k];
                  }
          }

           printf("%d
",max);
    }
    return 0;
}