16.10.7 A:4815. 【NOIP2016提高A组五校联考4】ksum

这道题目仔细想想就可以发现是一个堆，我们先做1~n的前缀和，然后我们把（1，n）放入堆，[a+1,b]和[a,b-1]才有可能成为下次取出的对
象，那么我就将[a+1,b]和[a,b-1]加进堆，这样重复 k 次即可,另外，如果每次都[a+1,b]和[a,b-1]加入堆，会得到重复的答案，那么，我们可以固定左端点。

反正c++优先队列爱怎么浪怎么堆怎么浪，就酱紫。

#include<cstdio>#include<queue>#define F(i,x,y) for(ll i=x;i<=y;i++)#define d(i,x,y) for(ll i=x;i>=y;i--)using namespace std;typedef long long ll;const ll maxn = 100000 + 200;ll a[maxn];ll sum[maxn];ll n,k;struct node{    ll l,r;    friend bool operator<(node x,node y)    {      return (sum[x.r] - sum[x.l-1]) < (sum[y.r] - sum[y.l-1]);    }}x;priority_queue<node> q;inline ll read(){    char ch = getchar();    ll data = 0;    while(ch<'0' || ch>'9')     ch = getchar();    while(ch>='0' && ch<='9')    {        data = data * 10 + ch - '0';        ch = getchar();    }    return data;} int main(){    freopen("ksum.in","r",stdin);    //freopen("ksum.out","w",stdout);    n = read(),k = read();    x.l = 1;    x.r = n;    sum[0] = 0;    F(i,1,n)     a[i] = read(),sum[i] = sum[i-1] + a[i];    q.push(x);    d(i,k,1)    {        x = q.top();        q.pop();        printf("%lld ",sum[x.r] - sum[x.l-1]);        if(x.r==n)        {            x.l++;            q.push(x);            x.l--;        }        x.r--;        q.push(x);    }    printf("\n");    return 0;}