[BZOJ]4003: [JLOI2015]城池攻占

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <stack>
#include <queue>
#include <cmath>
#include <set>
#include <map>
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define link(x) for(edge *j=h[x];j;j=j->next)
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
const int MAXN=3e5+10;
const double eps=1e-8;
#define ll long long
using namespace std;
struct edge{int t;edge*next;}e[MAXN<<1],*h[MAXN],*o=e;
void add(int x,int y){o->t=y;o->next=h[x];h[x]=o++;}
ll read(){
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch==‘-‘)f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-‘0‘,ch=getchar();
    return x*f;
}

typedef struct node{
    int ch[2],sz,dis,flag3;ll key,flag1,flag2;
}node;

node d[MAXN];
void newnode(int x,ll t){
    d[x].ch[0]=d[x].ch[1]=d[x].sz=d[x].dis=d[x].flag3=d[x].flag2=0;
    d[x].key=t;d[x].flag1=1;
}
void Flag1(int x,ll t){
    if(!x)return ;
    d[x].flag1*=t;d[x].flag2*=t;
    d[x].key*=t;
}

void Flag2(int x,ll t){
    if(!x)return ;
    d[x].key+=t;d[x].flag2+=t;
}
void Flag3(int x,int t){
    if(!x)return ;
    d[x].sz+=t;d[x].flag3+=t;
}

void push(int x){
    if(d[x].flag1!=1){
	if(d[x].ch[0])Flag1(d[x].ch[0],d[x].flag1);
	if(d[x].ch[1])Flag1(d[x].ch[1],d[x].flag1);
	d[x].flag1=1;
    }
    if(d[x].flag2){
	if(d[x].ch[0])Flag2(d[x].ch[0],d[x].flag2);
	if(d[x].ch[1])Flag2(d[x].ch[1],d[x].flag2);
	d[x].flag2=0;
    }
    if(d[x].flag3){
	if(d[x].ch[0])Flag3(d[x].ch[0],d[x].flag3);
	if(d[x].ch[1])Flag3(d[x].ch[1],d[x].flag3);
	d[x].flag3=0;
    }
}

int merge(int x,int y){
    if(!x||!y)return x+y;
    push(x);push(y);
    if(d[x].key>d[y].key)swap(x,y);
    d[x].ch[1]=merge(d[x].ch[1],y);
    if(d[d[x].ch[0]].dis<d[d[x].ch[1]].dis)swap(d[x].ch[0],d[x].ch[1]);
    d[x].dis=d[d[x].ch[1]].dis+1;
    return x;
}

int rt[MAXN];
ll H[MAXN],V[MAXN];
int op[MAXN];
int num[MAXN];

void dfs(int x){
    link(x){
	dfs(j->t);
	rt[x]=merge(rt[x],rt[j->t]);
    }
    while(rt[x]&&d[rt[x]].key<H[x]){
	num[x]++;
	push(rt[x]);
	rt[x]=merge(d[rt[x]].ch[0],d[rt[x]].ch[1]);
    }
    if(x!=1){
	if(op[x])Flag1(rt[x],V[x]);
	else Flag2(rt[x],V[x]);
    }
    if(rt[x])Flag3(rt[x],1);
}

int main(){
    int n=read();int m=read();
    inc(i,1,n)H[i]=read();
    int u;
    inc(i,2,n)u=read(),add(u,i),op[i]=read(),V[i]=read();
    ll k;
    inc(i,1,m)k=read(),newnode(i,k),u=read(),rt[u]=merge(rt[u],i);
    dfs(1);
    while(rt[1]){
	push(rt[1]);
	rt[1]=merge(d[rt[1]].ch[0],d[rt[1]].ch[1]);
    }
    inc(i,1,n)printf("%d\n",num[i]);
    inc(i,1,m)printf("%d\n",d[i].sz);
}