[洛谷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\n", ans); return 0; }