[JLOI2015]城池攻占

一、题目

点此看题

二、解法

先把每个人放到对应的点上，可以用左偏树，然后从下往上合并，每次就删除当前节点会牺牲的人，然后维护一个 加法$/$乘法 标记，时间复杂度$O(n\log n)$

#include <cstdio>
#include <iostream>
using namespace std;
#define int long long
const int M = 300005;
int read()
{
 int x=0,flag=1;char c;
 while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
 while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
 return x*flag;
}
int n,m,tot,f[M],h[M],fl[M],a[M],dep[M],a1[M],a2[M],c[M];
int val[M],dis[M],ls[M],rs[M],mul[M],add[M],rt[M];
struct edge
{
    int v,next;
    edge(int V=0,int N=0) : v(V) , next(N) {}
}e[2*M];
void upd(int x,int y,int z)
{
    if(!x) return ;
    val[x]*=y;val[x]+=z;
    mul[x]*=y;add[x]*=y;add[x]+=z;
}
void down(int x)
{
    upd(ls[x],mul[x],add[x]);
    upd(rs[x],mul[x],add[x]);
    mul[x]=1;add[x]=0;
}
int merge(int x,int y)
{
    if(!x || !y) return x|y;
    down(x);down(y);
    if(val[x]>val[y]) swap(x,y);
    rs[x]=merge(rs[x],y);
    if(dis[ls[x]]<dis[rs[x]]) swap(ls[x],rs[x]);
    dis[x]=dis[rs[x]]+1;
    return x;
}
void dfs(int u,int fa)
{
    dep[u]=dep[fa]+1;
    for(int i=f[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(v==fa) continue;
        dfs(v,u);
        rt[u]=merge(rt[u],rt[v]);
    }
    while(rt[u] && val[rt[u]]<h[u])
    {
        down(rt[u]);
        a1[u]++;a2[rt[u]]=dep[c[rt[u]]]-dep[u];
        rt[u]=merge(ls[rt[u]],rs[rt[u]]);
    }
    if(fl[u]) upd(rt[u],a[u],0);
    else upd(rt[u],1,a[u]);
}
signed main()
{
    dis[0]=-1;
    n=read();m=read();
    for(int i=1;i<=n;i++)
        h[i]=read();
    for(int i=2;i<=n;i++)
    {
        int j=read();fl[i]=read();a[i]=read();
        e[++tot]=edge(i,f[j]),f[j]=tot;
        e[++tot]=edge(j,f[i]),f[i]=tot;
    }
    for(int i=1;i<=m;i++)
    {
        val[i]=read();c[i]=read();
        mul[i]=1;
        rt[c[i]]=merge(rt[c[i]],i);
    }
    dfs(1,0);
    while(rt[1])
    {
        down(rt[1]);
        a2[rt[1]]=dep[c[rt[1]]];
        rt[1]=merge(ls[rt[1]],rs[rt[1]]);
    }
    for(int i=1;i<=n;i++) printf("%lld\n",a1[i]);
    for(int i=1;i<=m;i++) printf("%lld\n",a2[i]);
}