【题解】LuoGu2048：[NOI2010]超级钢琴

最新推荐文章于 2020-09-13 16:56:39 发布

丰川样子小孩姐

最新推荐文章于 2020-09-13 16:56:39 发布

阅读量205

点赞数 1

分类专栏：题解倍增优先队列（堆）文章标签：题解 LuoGu 堆倍增

本文链接：https://blog.csdn.net/ModestCoder_/article/details/108309533

版权

题解同时被 3 个专栏收录

435 篇文章

订阅专栏

倍增

12 篇文章

订阅专栏

优先队列（堆）

8 篇文章

订阅专栏

原题传送门
这道题目的id昭示着它的非同凡响
我可以把这题和别的几题结合起来，产生一种利用堆的节点最优思想
序列合并
 最小函数值
以上是两道和这题有共同思想的

本题是要求出前 $k$ 大的 $sum_j-sum_{i-1}$ 的和
对于每个节点 $i$ ，首先寻出 $[i + l - 1, i + r - 1]$ 区间内的 $p o s$ ，使得 $sum_{pos}-sum_{i-1}>=sum_j-sum_{i-1}(i+l-1<=j<=i+r-1)$
每个 $i$ 的 $p o s$ 用 $s t$ 表求出
那么每个节点 $i$ 目前最优的有了，把 $n$ 个节点最优的放到堆里面，每次取出堆里最大的，假设取出节点 $x$ ，对应最优解为 $y$ ，节点 $x$ 对应的区间为 $[l, r]$

因为从堆里取出来的，满足全局最优，所以统计到答案里面
又因为这是对于节点 $x$ 的最优，然后我们要找到最优之外次优的，把区间 $[l, r]$ 分成两半
$[l, y - 1] [y + 1, r]$ ，分别求出最优位置放到堆里即可

Code：

#include <bits/stdc++.h>
#define maxn 500010
using namespace std;
int sum[maxn], power[25], st[maxn][25], id[maxn][25], n, k, l, r;
struct node{
	int id, l, r, pos;
	bool operator < (const node &x) const{ return sum[x.pos] - sum[x.id - 1] > sum[pos] - sum[id - 1]; }
};
priority_queue <node> q;

inline int read(){
	int s = 0, w = 1;
	char c = getchar();
	for (; !isdigit(c); c = getchar()) if (c == '-') w = -1;
	for (; isdigit(c); c = getchar()) s = (s << 1) + (s << 3) + (c ^ 48);
	return s * w;
}

int calc(int l, int r){
	int ans = 0, x = -1e9;
	for (int i = 20; i >= 0; --i)
		if (l + power[i] - 1 <= r){
			if (st[l][i] > x) x = st[l][i], ans = id[l][i];
			l += power[i];
		}
	return ans;
}

int main(){
	n = read(), k = read(), l = read(), r = read();
	for (int i = 1; i <= n; ++i) st[i][0] = sum[i] = sum[i - 1] + read(), id[i][0] = i;
	power[0] = 1;
	for (int i = 1; i <= 20; ++i) power[i] = power[i - 1] << 1; 
	for (int j = 1; j <= 20; ++j)
		for (int i = 1; i + power[j] - 1 <= n; ++i){
			if (st[i][j - 1] > st[i + power[j - 1]][j - 1]) st[i][j] = st[i][j - 1], id[i][j] = id[i][j - 1];
			else st[i][j] = st[i + power[j - 1]][j - 1], id[i][j] = id[i + power[j - 1]][j - 1];
		}
	for (int i = 1; i <= n - l + 1; ++i){
		int L = i + l - 1, R = min(n, i + r - 1), x = calc(L, R);
		q.push((node){i, L, R, x});
	}
	long long ans = 0;
	while (k--){
		node tmp = q.top(); q.pop();
		ans += sum[tmp.pos] - sum[tmp.id - 1];
		if (tmp.pos > tmp.l) q.push((node){tmp.id, tmp.l, tmp.pos - 1, calc(tmp.l, tmp.pos - 1)});
		if (tmp.pos < tmp.r) q.push((node){tmp.id, tmp.pos + 1, tmp.r, calc(tmp.pos + 1, tmp.r)});
	}
	printf("%lld\n", ans);
	return 0;
}