[洛谷P1792][国家集训队]种树

题目大意:给出由$n$个数组成的环,取某个数就可以得到它的分数,相邻的两个数不能同时取。问取$m$个数可以得到的最大分数。

题解:建一个大根堆,贪心取,每个点记录前驱后继,取一个点就把前驱后继设成不能取;但这样是不正确的,但假设取了点$p$,可以把$pre_p,p,nxt_p$合成一个点,这个点的前驱为$pre_{pre_p}$,后继为$nxt_{nxt_p}$,价值为$V_{pre_p}+V_{nxt_p}-V_p$,取这个点代表取了$pre_p$和$nxt_p$。这样就可以保证正确

卡点:能信我$Error$打错了?

C++ Code:

#include <cstdio>
#include <queue>
#define maxn 200010
int n, m;
long long ans;
std::priority_queue<std::pair<int, int> > q;
int s[maxn], nxt[maxn], pre[maxn];
bool del[maxn];
int main() {
	scanf("%d%d", &n, &m);
	if (n < m << 1) {
		puts("Error!");
		return 0;
		puts("TANG.Yx");
	}
	for (int i = 1; i <= n; i++) {
		scanf("%d", s + i);
		q.push(std::make_pair(s[i], i));
		nxt[i] = i % n + 1;
		pre[i] = (i + n - 2) % n + 1;
	}
	for (int i = 1; i <= m; i++) {
		while (del[q.top().second]) q.pop();
		ans += q.top().first;
		int p = q.top().second, l = pre[p], r = nxt[p];
		q.pop();
		nxt[pre[p] = pre[l]] = pre[nxt[p] = nxt[r]] = p;
		del[l] = del[r] = true;
		s[p] = s[l] + s[r] - s[p];
		q.push(std::make_pair(s[p], p));
	}
	printf("%lld
", ans);
	return 0;
}

  

原文地址:https://www.cnblogs.com/Memory-of-winter/p/9744098.html