1. 程式人生 > >luoguP1792 種樹 貪心 雙向連結串列

luoguP1792 種樹 貪心 雙向連結串列

題目連結

這個題很早之前就出過 但是一直沒有去改

其實也不是很難,用雙向連結串列串起來整個環

每次貪心選取權值最大的

把它和它的前驅後繼縮成一個點 權值為前驅加後繼減去它的

仔細思考一下就是增加了一個反悔操作 用一個大根堆維護就好了

我一開始sbsbsetset去搞 發現過載運算子的時候
如果你用來比較的值是相同的話 意思兩個編號權值相同

那麼就會有一個點被setset的去重功能刪了

所以以後打過載運算子還是考慮一下另一維相等的情況把…

這樣子就是對的了

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;
}