题目:
思路:
反悔贪心,将所有坑都放在优先队列中。
先从值最大的开始贪,在贪的时候还要记录它的左右,因为存在一种情况,如:
处理左右位置不可选的坑:
代码:
#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;
}