bzoj 4003 城池攻占 （可并堆（左偏树））

小铭铭最近获得了一副新的桌游，游戏中需要用 m 个骑士攻占 n 个城池。

这 n 个城池用 1 到 n 的整数表示。除 1 号城池外，城池 i 会受到另一座城池 fi 的管辖，

其中 fi <i。也就是说，所有城池构成了一棵有根树。这 m 个骑士用 1 到 m 的整数表示，其

中第 i 个骑士的初始战斗力为 si，第一个攻击的城池为 ci。

每个城池有一个防御值 hi，如果一个骑士的战斗力大于等于城池的生命值，那么骑士就可

以占领这座城池；否则占领失败，骑士将在这座城池牺牲。占领一个城池以后，骑士的战斗力

将发生变化，然后继续攻击管辖这座城池的城池，直到占领 1 号城池，或牺牲为止。

除 1 号城池外，每个城池 i 会给出一个战斗力变化参数 ai;vi。若 ai =0，攻占城池 i 以后骑士战斗力会增加 vi；若 ai =1，攻占城池 i 以后，战斗力会乘以 vi。注意每个骑士是单独计算的。也就是说一个骑士攻击一座城池，不管结果如何，均不会影响其他骑士攻击这座城池的结果。

现在的问题是，对于每个城池，输出有多少个骑士在这里牺牲；对于每个骑士，输出他攻占的城池数量。

Input

第 1 行包含两个正整数 n;m，表示城池的数量和骑士的数量。

第 2 行包含 n 个整数，其中第 i 个数为 hi，表示城池 i 的防御值。

第 3 到 n +1 行，每行包含三个整数。其中第 i +1 行的三个数为 fi;ai;vi，分别表示管辖

这座城池的城池编号和两个战斗力变化参数。

第 n +2 到 n + m +1 行，每行包含两个整数。其中第 n + i 行的两个数为 si;ci，分别表

示初始战斗力和第一个攻击的城池。

Output

 输出 n + m 行，每行包含一个非负整数。其中前 n 行分别表示在城池 1 到 n 牺牲的骑士

数量，后 m 行分别表示骑士 1 到 m 攻占的城池数量。

Sample Input

5 5 50 20 10 10 30 1 1 2 2 0 5 2 0 -10 1 0 10 20 2 10 3 40 4 20 4 35 5

Sample Output

2 2 0 0 0 1 1 3 1 1

思路： 我们对于每个一个树节点可以维护一个堆，堆里放现在还存活的骑士，我们在dfs 回退的时候，就可以合并这个节点的孩子的堆，合并完之后检查是否存在在这个节点要挂掉的骑士，这样的话，就可以pop掉了。并且检查完挂掉的骑士后，我们就需要对堆里的骑士进行升级了，那么暴力是不存在的，所以我们可以打上乘和加两个标记，这里为甚么打标记是对的，因为这里的标记并不会影响到堆的性质，我们维护的最小堆，所以给堆里的每一个元素乘一个整数，或者加一个正负数，都不会影响堆里元素的相对关系的，所以这里是可行的。

代码：

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
const int N =300005;
struct merge_dui
{
    int l,r;
    int dis;
    ll val;
    ll cheng,jia;
}T[N];
int root[N];
//int sz[N]; /// 看情况用

int n,m;
ll def[N];
int op[N];
ll val[N];
int beg[N];
int End[N];
int dep[N];
int dead[N];
vector<int >ve[N];

void Biao(int rt,ll cheng ,ll jia)
{
    if(rt==0) return ;
    T[rt].val*=cheng;
    T[rt].val+=jia;
    T[rt].cheng*=cheng; T[rt].jia*=cheng; T[rt].jia+=jia;
}

void push_down(int rt)
{
    ll cheng=T[rt].cheng; ll jia=T[rt].jia;
    Biao(T[rt].l,cheng,jia);
    Biao(T[rt].r,cheng,jia);
    T[rt].cheng=1; T[rt].jia=0;
}

int Merge(int r1,int r2)   /// merge 后返回一个根节点
{
    if(r1==0||r2==0) return r1+r2;
    push_down(r1); push_down(r2);
    if(T[r1].val > T[r2].val )swap(r1,r2);  /// 维护一个最小堆
    T[r1].r=Merge(T[r1].r,r2);
    if(T[T[r1].l].dis<T[T[r1].r].dis) swap(T[r1].l,T[r1].r);
    T[r1].dis=T[T[r1].r].dis+1;
    return r1;
}

int Pop(int rt) /// pop 后返回一个根节点
{
    return Merge(T[rt].l,T[rt].r);
}

void dfs(int u,int deep)
{
    dep[u]=deep;
    for(int i=0;i<ve[u].size();i++){
        int v=ve[u][i];
        dfs(v,deep+1);
    }

    for(int i=0;i<ve[u].size();i++){
        int v=ve[u][i];
        root[u]=Merge(root[u],root[v]);
    }

    while(root[u]!=0&&T[root[u]].val<def[u]){
        push_down(root[u]);
        dead[u]++;
        End[root[u]]=u;
        root[u]=Pop(root[u]);
    }
    if(op[u]==0) Biao(root[u],1,val[u]);
    else Biao(root[u],val[u],0);
}

int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%lld",&def[i]);
    }
    int fa;
    op[1]=0; val[1]=0;
    for(int i=2;i<=n;i++){
        scanf("%d %d %lld",&fa,&op[i],&val[i]);
        ve[fa].push_back(i);
    }
    for(int i=1;i<N;i++) T[i].cheng=1,T[i].jia=0;
    for(int i=1;i<=m;i++){
        scanf("%lld %d",&T[i].val,&beg[i]);  /// 先把每个骑士当做一个堆，如果他们攻击的城堡相同，并成一个堆。
        root[beg[i]]=Merge( root[beg[i]],i );
    }

    dfs(1,1);
    for(int i=1;i<=n;i++){
        printf("%d\n",dead[i]);
    }
    for(int i=1;i<=m;i++){
        printf("%d\n",dep[beg[i]]-dep[End[i]]);
    }

    return 0;
}