luoguP1792 種樹 貪心 雙向連結串列
阿新 • • 發佈:2019-01-22
這個題很早之前就出過 但是一直沒有去改
其實也不是很難,用雙向連結串列串起來整個環
每次貪心選取權值最大的
把它和它的前驅後繼縮成一個點 權值為前驅加後繼減去它的
仔細思考一下就是增加了一個反悔操作 用一個大根堆維護就好了
我一開始用去搞 發現過載運算子的時候
如果你用來比較的值是相同的話 意思兩個編號權值相同
那麼就會有一個點被的去重功能刪了
所以以後打過載運算子還是考慮一下另一維相等的情況把…
這樣子就是對的了
struct node { int val, id; bool operator < (const node &T) const { if(val == T.val) return id < T.id; return val > T.val; } }a[N];
Codes
#include<bits/stdc++.h> using namespace std; const int N = 2e5 + 10; int n, m, ans, del[N]; int from[N], to[N]; struct node { int val, id; bool operator < (const node &T) const { if(val == T.val) return id < T.id; return val > T.val; } }a[N]; set<node> T; void find() { while(del[(*T.begin()).id]) T.erase(T.begin()); node it = *T.begin(); T.erase(it); int x = it.id; ans += a[x].val; a[x].val = a[from[x]].val + a[to[x]].val - a[x].val; del[from[x]] = del[to[x]] = 1; to[from[from[x]]] = x; from[to[to[x]]] = x; from[x] = from[from[x]]; to[x] = to[to[x]]; T.insert(a[x]); } int main() { #ifndef ONLINE_JUDGE freopen("2151.in", "r", stdin); freopen("2151.out", "w", stdout); #endif scanf("%d%d", &n, &m); if(m > n / 2) return puts("Error!"), 0; for(int i = 1; i <= n; ++ i) { scanf("%d", &a[i].val); from[i] = i - 1; to[i] = i + 1; a[i].id = i; T.insert(a[i]); } from[1] = n; to[n] = 1; for(int i = 1; i <= m; ++ i) find(); cout << ans << endl; return 0; }