[左偏树+lazy思想]BZOJ 4003——[JLOI2015]城池攻占

题目梗概

一棵有根树，每一个节点都是一个城池，有一个防御值si。

有若干个战士，第i 个战士初始攻击力为 ai，初始攻击的城池为 ci。

如果攻击力大于等于城池的防御值，则攻击成功，战士继续攻击父节点，否则攻击失败。

攻击完第i 个城市后战士的攻击力会加上一个数或者乘以一个数（每一个城池属性不同）。

每一个战士都是独立的，问每一个战士攻占了几个城池。

解题思路

先来说说大致思想，我们考虑从叶子节点开始处理，在每个节点的所有战士形成一个块，对于战斗力小于该叶子节点防御值的战士，我们将其删除，对剩下的战士的战斗力我们进行相应的加操作或乘操作，接着和该节点的父亲合并。

既能快速获取最小值，又能合并，首选就是左偏树吧。

因为要加操作或乘操作，所以我们要利用Lazy思想。

为了方便处理，我们假设1节点的父亲是0，且0节点的防御值为正无穷。

#include<cstdio>
#include<algorithm>
#define LL long long
using namespace std;
const int maxn=300005;
inline LL _read(){
    LL num=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
    return num*f;
}
struct jz{
    int id,son[2],s;
    LL x,tag1,tag2;
}t[maxn];
int n,m,f[maxn],a[maxn],ro[maxn],c[maxn],tot,dep[maxn],die[maxn],ans[maxn];
LL w[maxn],v[maxn],s[maxn];
int add(LL x,int y){tot++;t[tot].x=x;t[tot].id=y;t[tot].s=1;t[tot].tag2=1;return tot;}
void Pushdown(int x){
    if (!x) return;
    LL tag1=t[x].tag1,tag2=t[x].tag2;
    int son0=t[x].son[0],son1=t[x].son[1];
    t[x].tag1=0;t[x].tag2=1;
    t[son0].x=t[son0].x*tag2+tag1;t[son1].x=t[son1].x*tag2+tag1;
    t[son0].tag2*=tag2;t[son0].tag1=t[son0].tag1*tag2+tag1;
    t[son1].tag2*=tag2;t[son1].tag1=t[son1].tag1*tag2+tag1;
}
int merge(int x,int y){
    Pushdown(x);Pushdown(y);
    if (x==0||y==0) return x|y;
    if (t[x].x>t[y].x) swap(x,y);
    t[x].son[1]=merge(t[x].son[1],y);
    if (t[t[x].son[1]].s>t[t[x].son[0]].s) swap(t[x].son[0],t[x].son[1]);
    t[x].s=t[t[x].son[0]].s+t[t[x].son[1]].s+1;
    return x;
}
void get(int x){ro[x]=merge(t[ro[x]].son[0],t[ro[x]].son[1]);}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    n=_read(),m=_read();
    for (int i=1;i<=n;i++) w[i]=_read();
    for (int i=2;i<=n;i++) f[i]=_read(),a[i]=_read(),v[i]=_read();
    for (int i=1;i<=m;i++){
        s[i]=_read(),c[i]=_read();
        ro[c[i]]=merge(ro[c[i]],add(s[i],i));
    }
    w[0]=9223372036854775807;for (int i=1;i<=n;i++) dep[i]=dep[f[i]]+1;
    for (int i=n;i>=0;i--){
        while (ro[i]!=0&&t[ro[i]].x<w[i]){die[t[ro[i]].id]=i;get(i);ans[i]++;}
        if (ro[i]){Pushdown(ro[i]);if (a[i]) t[ro[i]].x*=v[i],t[ro[i]].tag2=v[i];else t[ro[i]].x+=v[i],t[ro[i]].tag1=v[i];}
        if (i) ro[f[i]]=merge(ro[f[i]],ro[i]);
    }
    for (int i=1;i<=n;i++) printf("%d\n",ans[i]);
    for (int i=1;i<=m;i++) printf("%d\n",dep[c[i]]-dep[die[i]]);
    return 0;
}