左偏树:可以在lon(n)时间内合并两个堆
学习博客:https://www.cnblogs.com/ShuraEye/p/11379618.html
P3377 【模板】左偏树(可并堆)
模板题,注意并查集路径压缩
#include<bits/stdc++.h>
#define ll long long
#define N 100010
using namespace std;
int n,m,dis[N],l[N],r[N],v[N],vis[N],f[N],a[N],tot=0;
void init(int x){
v[++tot]=x;
l[tot]=r[tot]=dis[tot]=0;
}
int find(int x){
if(f[x]!=x)f[x]=find(f[x]);
return f[x];
}
int merge(int x,int y){
if(!x||!y)return x+y;
if(v[x]>v[y]||(v[x]==v[y]&&x>y))swap(x,y);
r[x]=merge(r[x],y);
if(dis[l[x]]<dis[r[x]])swap(l[x],r[x]);
dis[x]=dis[r[x]]+1;
return x;
}
int top(int x){
return v[x];
}
void pop(int x){
vis[x]=1;
//f[l[x]]=l[x];f[r[x]]=r[x];
f[l[x]]=f[r[x]]=f[x]=merge(l[x],r[x]);
}
int main(){
scanf("%d%d",&n,&m);
dis[0]=-1;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]),init(a[i]);
for(int i=1;i<=n;i++)f[i]=i;
int op,x,y;
while(m--){
scanf("%d%d",&op,&x);
if(op==1){
scanf("%d",&y);
int fx=find(x),fy=find(y);
if(fx!=fy&&!vis[x]&&!vis[y]){
f[fx]=f[fy]=merge(fx,fy);
}
}
else{
if(vis[x])printf("-1\n");
else{
x=find(x);
printf("%d\n",top(x));
pop(x);
}
}
}
return 0;
}
P3261 [JLOI2015]城池攻占(带懒标记的左偏树)
在树上建立左偏树是要区分树的节点,和每个节点的左偏树中的点
考虑dfs后从叶子往根递推。对于处理每个节点时,首先要把从所有孩子爬上来的一大坨士兵整合成一坨,然后把里面小于h的全部挑出来。很容易想到要用可并堆,维护每个节点孩子爬上来的士兵的攻击力最小值。
所以每个节点建一颗左偏树,初始时把所有在那里的士兵塞进去,每次处理时一直pop,更新在这个节点死掉的士兵和这个士兵爬的层数,直到堆顶≥h。更新这些士兵能力值时,在根节点上打一个乘法tag和一个加法tag就行了。记得pushdown的时候加法tag也要乘上父亲的乘法tag。
复杂度:初始插入所有士兵为mlogm。每个士兵最多被带上n层,为nm。每个士兵只会被弹出一次,mlogm。所以总复杂度O(mn+mlogm)。
#include<bits/stdc++.h>
#define ll long long
#define N 1000010
using namespace std;
int n,m;
vector<int>vec[N];
ll a[N],v[N],l[N],r[N],rt[N],dep[N],ans[N],ans1[N],dis[N],h[N],pos[N];
ll mul[N],add[N],val[N];
void pushdown(int x){
if(l[x]){
add[l[x]]=add[l[x]]*mul[x]+add[x];
mul[l[x]]*=mul[x];
val[l[x]]=val[l[x]]*mul[x]+add[x];
}
if(r[x]){
add[r[x]]=add[r[x]]*mul[x]+add[x];
mul[r[x]]*=mul[x];
val[r[x]]=val[r[x]]*mul[x]+add[x];
}
mul[x]=1;add[x]=0;
}
int merge(int x,int y){
if(!x||!y)return x+y;
pushdown(x);pushdown(y);
if(val[x]>val[y])swap(x,y);
r[x]=merge(r[x],y);
if(dis[l[x]]<dis[r[x]])swap(l[x],r[x]);
dis[x]=dis[r[x]]+1;
return x;
}
int del(int x){
pushdown(x);
return merge(l[x],r[x]);
}
void dfs(int x,int fa){
for(int to:vec[x]){
if(to==fa)continue;
dep[to]=dep[x]+1;
dfs(to,x);
rt[x]=merge(rt[x],rt[to]);
}
while(val[rt[x]]<h[x]&&rt[x]){
ans[x]++,ans1[rt[x]]=x;
rt[x]=del(rt[x]);
}
if(a[x]==1)mul[rt[x]]*=v[x],add[rt[x]]=add[rt[x]]*v[x],val[rt[x]]*=v[x];
else add[rt[x]]+=v[x],val[rt[x]]+=v[x];
// pushdown(rt[x]);
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)scanf("%lld",&h[i]);
int x,y;
for(int i=2;i<=n;i++){
scanf("%d%lld%lld",&x,&a[i],&v[i]);
vec[x].push_back(i);
}
dis[0]=-1;dep[1]=1;
for(int i=1;i<=m;i++)mul[i]=1,add[i]=0,dis[i]=0;
for(int i=1;i<=m;i++){
scanf("%lld%lld",&val[i],&pos[i]);
rt[pos[i]]=merge(rt[pos[i]],i);
}
dfs(1,0);dep[0]=0;
while(rt[1]){
ans1[rt[1]]=0;
rt[1]=del(rt[1]);
}
for(int i=1;i<=n;i++)printf("%lld\n",ans[i]);
// for(int i=1;i<=m;i++)cout<<ans1[i]<<" ";
// cout<<endl;
for(int i=1;i<=m;i++)printf("%lld\n",dep[pos[i]]-dep[ans1[i]]);
return 0;
}
P1552 [APIO2012]派遣
从叶子节点开始,每个点开始是一个大根堆,堆里的就是这个点的人。
往父亲那里合并堆,记录堆的大小,费用的总和。
从儿子合并完毕后,在每个节点,不断踢出费用最大的人,直到费用的总和<=m 这就是这个点的最优方案了。(显然,花费最小的都留下了)
#include<bits/stdc++.h>
#define N 100100
#define ll long long
using namespace std;
vector<int>vec[N];
int n,m;
int l[N],r[N],val[N],num[N],a[N],rt[N],dis[N];
ll sum[N],ans=0;
int merge(int x,int y){
if(!x||!y)return x+y;
if(val[x]<val[y])swap(x,y);
r[x]=merge(r[x],y);
if(dis[l[x]]<dis[r[x]])swap(l[x],r[x]);
sum[x]=sum[l[x]]+sum[r[x]]+val[x];
num[x]=num[l[x]]+num[r[x]]+1;
dis[x]=dis[r[x]]+1;
return x;
}
void dfs(int x){
for(int i=0;i<vec[x].size();i++){
int to=vec[x][i];
dfs(to);
rt[x]=merge(rt[x],rt[to]);
}
while(sum[rt[x]]>m&&rt[x]){
rt[x]=merge(l[rt[x]],r[rt[x]]);
}
if(sum[rt[x]]<=m)ans=max(ans,(ll)a[x]*num[rt[x]]);
}
int main(){
scanf("%d%d",&n,&m);
int x;
for(int i=1;i<=n;i++){
scanf("%d%d%d",&x,&val[i],&a[i]);
vec[x].push_back(i);
rt[i]=i;num[i]=1;sum[i]=val[i];
}
dis[0]=-1;
dfs(1);
printf("%lld",ans);
return 0;
}