NOIP模拟题 订餐

题目描述
这个周末，Joe 一如既往的拿出了外卖的点餐单。菜单上从上到下列着 N 个菜，
每个菜 Joe 会列出一个美味值。Joe 这次希望从菜单上连续地点 K 个菜，并且美
味值之和最大。
但在计算美味值之和时，Joe 有自己的一套计算方法：由于他会按照顺序享受这
K 个菜，并且越吃越后面的菜他就越享受，所以对于他吃的第 i 个菜，设它的美
味值为 D，美味值之和就会加上 D*i。例如，如果他要顺序吃美味值为 3, 8, 5 的
菜，那么美味值之和为 3 + 8*2 + 5*3 = 34。
Joe 给出了这 N 个菜的美味值，他想知道所能达到最大的美味值之和。

输入
第一行两个整数 N，K，代表菜的总数和他要选择的菜数。
第二行 N 个整数，代表从上到下菜单上菜品的美味值。

输出
一行一个整数，代表最大的美味值之和。

样例
5 3
2 4 1 5 3
21

数据范围
对于 40%的数据，N,K<=2000;
对于 100%的数据，N,K<=100000，各菜品的美味值 D 满足 0<=D<=1000.

用一个前缀和，先处理出以1开头的美味值，然后用前缀和往后模拟就可以了。
看图

计算出第一段，然后计算第二段时，可以用第一段的值-第一段的和+a[4]*3

时间复杂度：O(n)

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define LL long long
using namespace std;
const int N=100009;
int n,k;
LL a[N],s[N],b[N],ans;
int main()
{
    freopen("dish.in","r",stdin);
    freopen("dish.out","w",stdout);
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]),s[i]=s[i-1]+a[i];

    for(int i=1;i<=k;i++) b[1]+=a[i]*1ll*i;

    for(int i=2;i<=n-k+1;i++)
    {
        b[i]=b[i-1]-(s[i-1+k-1]-s[i-2])+a[i+k-1]*1ll*k;
    }
    //for(int i=1;i<=n;i++) printf("%lld ",b[i]);puts("");
    for(int i=1;i<=n;i++) ans=max(ans,b[i]);
    printf("%lld\n",ans);
    return 0;
}