【题解】LuoGu1792：[国家集训队]种树

丰川样子小孩姐

于 2020-08-20 15:38:02 发布

阅读量219

点赞数 1

分类专栏：题解 LuoGu 贪心文章标签：题解 LuoGu 贪心

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

版权

题解同时被 3 个专栏收录

435 篇文章

订阅专栏

LuoGu

255 篇文章

订阅专栏

贪心

19 篇文章

订阅专栏

本文探讨了在一个环形结构中应用贪心算法解决特定问题的实现细节。通过定义节点和堆结构，利用优先队列进行操作，实现了对环形链表的高效处理。在每次迭代中选择当前价值最大的节点进行合并，直至达到预定的合并次数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题传送门
 在这道题的基础上变成了一个环

Code：

#include <bits/stdc++.h>
#define maxn 500010
#define LL long long
using namespace std;
struct heap{
	int num;
	LL val;
	bool operator < (const heap &x) const{return x.val > val;}
};
priority_queue <heap> q;
struct node{
	int l, r;
	LL val;
}a[maxn];
int n, m, vis[maxn];

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 main(){
	n = read(), m = read();
	if (m > (n >> 1)) return puts("Error!"), 0;
	for (int i = 1; i <= n; ++i){
		a[i].val = read(), a[i].l = i - 1, a[i].r = i + 1;
		q.push((heap){i, a[i].val});
	}
	a[1].l = n, a[n].r = 1;
	LL ans = 0;
	while (!q.empty()){
		heap tmp = q.top(); q.pop();
		int x = tmp.num;
		if (vis[x]) continue;
		ans += tmp.val, --m;
		if (!m) break;
		a[x].val = a[a[x].l].val + a[a[x].r].val -  a[x].val;
		vis[a[x].l] = vis[a[x].r] = 1;
		q.push((heap){x, a[x].val});
		a[x].l = a[a[x].l].l, a[x].r = a[a[x].r].r;
		a[a[x].l].r = x, a[a[x].r].l = x;
	}
	printf("%lld\n", ans);
	return 0;
}