P1484 种树（反悔贪心，priority）

Looy_cai

已于 2022-04-15 21:44:13 修改

阅读量646

点赞数

CC 4.0 BY-SA版权

分类专栏：动态规划思维 dp 文章标签：算法

于 2022-04-14 20:14:13 首次发布

本文链接：https://blog.csdn.net/lalala625/article/details/124180019

思维同时被 3 个专栏收录

68 篇文章

订阅专栏

19 篇文章

订阅专栏

动态规划

5 篇文章

订阅专栏

这篇博客介绍了如何运用反悔贪心策略解决最大价值种树问题。通过维护一个优先队列，从最大值开始种植，同时考虑左右节点的影响，不断更新树的价值并排除已选择的节点，直至达到限制的树的数量。代码中详细展示了C++的实现过程。

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

LINK

题目：

在这里插入图片描述

思路：

反悔贪心，将所有坑都放在优先队列中。
先从值最大的开始贪，在贪的时候还要记录它的左右，因为存在一种情况，如：
在这里插入图片描述
处理左右位置不可选的坑：

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
const int N=5e5+10;
const int inf=0x3f3f3f3f;
struct node{
	int a,pos;
	//priority_queue中自定义的比较函数的效果和sort()是相反的
	bool operator < (const node &x) const{
		return a<x.a;
	}
};
priority_queue<node>q;
int vis[N],l[N],r[N],a[N];
ll ans;
inline void del(int x){
	r[l[x]]=r[x];
	l[r[x]]=l[x];
}
signed main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	int n,k;cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		l[i]=i-1;r[i]=i+1;//左右位置 
		q.push((node){a[i],i});
	}
	//从大到小 
	while(q.top().a>0&&k){//若值<0则不再种/可种数量超则不再种 
		node x=q.top();q.pop();
		if(vis[x.pos])continue;//访问过了 
		k--;//树减1  
		ans+=x.a; //种这棵树 
		int y=x.pos;//此树位置 
		int ls=l[y],rr=r[y];//此树左右两边的位置 
		//把此树左右两边加起来的值-此树的值加入队列中，为另一种选择 
		int w=-x.a ;
		if(ls)w+=a[ls];
		if(rr<=n)w+=a[rr];
		a[y]=w;
		q.push((node){w,y}); //将此点存入队列中
		vis[ls]=vis[rr]=1;//标记，不能再选这两个位置
		if(ls)del(ls);//更新这点的左右位置 
		if(rr<=n)del(rr);
	}
	cout<<ans<<endl;
	return 0;
}

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
const int N=5e5+10;
const int inf=0x3f3f3f3f;
struct node{
	int a,pos;
	//priority_queue中自定义的比较函数的效果和sort()是相反的
	bool operator < (const node &x) const{
		return a<x.a;
	}
};
priority_queue<node>q;
int vis[N],l[N],r[N],a[N];
ll ans;
signed main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	int n,k;cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		l[i]=i-1;r[i]=i+1;//左右位置 
		q.push((node){a[i],i});
	}
	//从大到小 
	while(q.top().a>0&&k){//若值<0则不再种/可种数量超则不再种 
		node x=q.top();q.pop();
		if(vis[x.pos])continue;//访问过了 
		k--;//树减1  
		ans+=x.a; //种这棵树 
		int y=x.pos;//此树位置 
		int ls=l[y],rr=r[y];//此树左右两边的位置 
		//把此树左右两边加起来的值-此树的值加入队列中，为另一种选择 
		int w=-x.a ;
		if(ls)w+=a[ls];
		if(rr<=n)w+=a[rr];
		a[y]=w;
		q.push((node){w,y}); //将此点存入队列中
		vis[ls]=vis[rr]=1;//标记，不能再选这两个位置
		if(ls)l[y]=l[l[y]],r[l[y]]=y;//更新这点的左右位置 
		if(rr<=n)r[y]=r[r[y]],l[r[y]]=y;
	}
	cout<<ans<<endl;
	return 0;
}