【JZOJ4815】ksum【堆】

题目：

题目链接：https://jzoj.net/senior/#main/show/4815
求一个数列前$k$大的字段和。

---

思路：

由于${\sum }_{i=k}^{n}a[i]<{\sum }_{i=k}^{n+1}a[i]$，所以我们直接建立一个大根堆，装对于每一个数$a[i]$的$sum(i\sim last[i])$，其中$last[i]$初始均为$n$。
然后从对顶弹出最大值，将$last[i]--$，再次插入。
时间复杂度$O(q\log n)$

---

代码：

#include <queue>
#include <cstdio>
#include <algorithm>
#define mp make_pair
using namespace std;
typedef long long ll;

const int N=100010;
int n,m,last[N];
ll a[N],sum[N];
priority_queue<pair<ll,int> > q;

int main()
{
	freopen("ksum.in","r",stdin);
	freopen("ksum.out","w",stdout);
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
		sum[i]=sum[i-1]+a[i];
		last[i]=n;
	}
	for (int i=1;i<=n;i++)
		q.push(mp(sum[n]-sum[i-1],i));
	while (m--)
	{
		printf("%lld ",q.top().first);
		int pos=q.top().second;
		q.pop();
		if (last[pos]>pos)
		{
			last[pos]--;
			q.push(mp(sum[last[pos]]-sum[pos-1],pos));
		}
	}
	return 0;
}