正题
题目链接:https://jzoj.net/senior/#main/show/4815
题目大意
nnn个数的序列,求前kkk大的子段。
解题思路
一个区间的答案肯定比他的所有子区间要大,所以我们先把(1,i)(1,i)(1,i)都丢进堆里,然后每次取出最大的再将(l+1,r)(l+1,r)(l+1,r)丢进去即可。
codecodecode
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define ll long long
using namespace std;
const ll N=1e5+10;
struct node{ll l,r,val;
};
bool operator<(node x,node y)
{return x.val<y.val;}
ll n,k,a[N];
priority_queue<node> q;
int main()
{freopen("ksum.in","r",stdin);freopen("ksum.out","w",stdout); scanf("%lld%lld",&n,&k);for(ll i=1;i<=n;i++){scanf("%lld",&a[i]),a[i]+=a[i-1];q.push((node){1,i,a[i]});}for(ll i=1;i<=k;i++){node x=q.top();q.pop();printf("%lld ",x.val);q.push((node){x.l+1,x.r,a[x.r]-a[x.l]});}
}