JZOJ4815. 【NOIP2016提高A组五校联考4】

Description

Input

Output

Sample Input

样例输入1：

3 4
1 3 4

样例输入2：

3 3
10 2 7

Sample Output

样例输出1：

8 7 4 4

样例输出2：

19 12 10

Data Constraint

Hint

分析

因为N非常大，
所以我们不可能将所有的子串枚举出来。

其实，我们只需要找到前K个子串就可以了。，
有一些子串是可以不用枚举的。

因为所有的数都是正整数，
所以最大的子串就一定是全部加起来，

那么，第二大的呢？
要么去掉最左边的一个，要么去掉最右边的一个。

我们可以得到一个结论：
一个子串[l,r]是当前最大的，
那么，下一次再选最大的时候才有可能选到[l+1,r]和[l,r-1]

我们就建一个堆，先放入[1,n]
然后取出最大的[l,r]，
然后在加入[l+1,r],[l,r-1]
只要判断一下重复就可以了。

code(c++)

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h> 
#include<queue>
using namespace std;
struct note{int l,r;long long s;}; 
priority_queue <note> q;
int n,k,a[100003];
long long s;
note t,z;
bool operator <(note a,note b){return a.s<b.s;}
int main()
{
    freopen("ksum.in","r",stdin);
    freopen("ksum.out","w",stdout);
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        s+=a[i];
    }
    t.s=s;t.l=1;t.r=n;
    q.push(t);
    for(int i=1;i<=k;i++)
    {
        t=q.top();
        q.pop();
        printf("%lld ",t.s);
        if(t.l!=t.r)
        {
            z=t;z.r--;z.s-=a[t.r];
            q.push(z);
            if(t.r==n)
            {
                z=t;z.l++;z.s-=a[t.l];
                q.push(z);
            }
        }
    }
}