洛谷P3261 [JLOI2015]城池攻占（左偏树）

传送门

每一个城市代表的点开一个小根堆，把每一个骑士合并到它开始攻占的城池所代表的点上

然后开始dfs，每一次把子树里那些还活着的骑士合并上来

然后再考虑当前点的堆，一直pop直到骑士全死光或者剩下的骑士的攻击力都大于等于当前城池的生命值，同时维护城池和骑士的答案

然后修改的话在堆顶打一个标记，需要的时候下传即可

//minamoto
#include<bits/stdc++.h>
#define ll long long
using namespace std;
#define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[1<<21],*p1=buf,*p2=buf;
template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
inline ll read(){
    #define num ch-'0'
    char ch;bool flag=0;ll res;
    while(!isdigit(ch=getc()))
    (ch=='-')&&(flag=true);
    for(res=num;isdigit(ch=getc());res=res*10+num);
    (flag)&&(res=-res);
    #undef num
    return res;
}
char sr[1<<21],z[20];int K=-1,Z;
inline void Ot(){fwrite(sr,1,K+1,stdout),K=-1;}
inline void print(int x){
    if(K>1<<20)Ot();if(x<0)sr[++K]=45,x=-x;
    while(z[++Z]=x%10+48,x/=10);
    while(sr[++K]=z[Z],--Z);sr[++K]='
';
}
const int N=3e5+5;
int head[N],Next[N],ver[N],tot;
inline void add_edge(int u,int v){
    ver[++tot]=v,Next[tot]=head[u],head[u]=tot;
}
int L[N],R[N],rt[N],d[N],ans1[N],ans2[N],dep[N],c[N],n,m;
ll add[N],mul[N],v[N],h[N],val[N];bool fl[N];
inline void ppd(int p,ll m,ll a){
    if(p){
        v[p]*=m,v[p]+=a;
        mul[p]*=m,add[p]*=m,add[p]+=a;
    }
}
inline void pd(int p){
    ppd(L[p],mul[p],add[p]),ppd(R[p],mul[p],add[p]);
    mul[p]=1,add[p]=0;
}
int merge(int x,int y){
    if(!x||!y) return x+y;
    pd(x),pd(y);
    if(v[x]>v[y]) swap(x,y);
    R[x]=merge(R[x],y);
    if(d[L[x]]<d[R[x]]) swap(L[x],R[x]);
    d[x]=d[R[x]]+1;return x;
}
void dfs(int u,int fa){
    dep[u]=dep[fa]+1;
    for(int i=head[u];i;i=Next[i]){int v=ver[i];dfs(v,u),rt[u]=merge(rt[u],rt[v]);}
    while(rt[u]&&v[rt[u]]<h[u]){
        pd(rt[u]);++ans1[u],ans2[rt[u]]=dep[c[rt[u]]]-dep[u];
        rt[u]=merge(L[rt[u]],R[rt[u]]);
    }
    fl[u]?ppd(rt[u],val[u],0):ppd(rt[u],1,val[u]);
}
int main(){
//    freopen("testdata.in","r",stdin);
    n=read(),m=read();
    for(int i=1;i<=n;++i) h[i]=read();
    for(int i=2;i<=n;++i){
        int fa=read();fl[i]=read(),val[i]=read();
        add_edge(fa,i);
    }
    for(int i=1;i<=m;++i){
        v[i]=read(),c[i]=read();
        mul[i]=1;
        rt[c[i]]=merge(rt[c[i]],i);
    }
    dfs(1,0);
    while(rt[1]) pd(rt[1]),ans2[rt[1]]=dep[c[rt[1]]],rt[1]=merge(L[rt[1]],R[rt[1]]);
    for(int i=1;i<=n;++i) print(ans1[i]);
    for(int i=1;i<=m;++i) print(ans2[i]);
    Ot();
    return 0;
}