bzoj4003[JLOI2015]城池攻占

bzoj4003[JLOI2015]城池攻占

题意：

有n个城池组成根节点为1的树，m个人，当一个人的战斗力大于等于攻打城市的防御力，就能攻占这个城市，来到这个城市的父节点，否则该人会牺牲在这个城市。当一个城市被攻占时，会使攻占的人的战斗力加或乘上某个数。现在给出m个人的最开始攻打的城市和初始战斗力，求在每个城市的牺牲人数和每个人一共攻打几个城市。注意这m个人处在不同的时空，即攻击互不影响，且每个人会一直往上攻打除非牺牲或到达根节点。

题解：

由于对一些数乘一个正数或加一个数这些数的相对大小不变，故可以用可并堆。每个人作为堆的一个节点，然后dfs，对每个节点将所有其子节点存储的堆合并，把牺牲的节点弹出，并对剩余的节点打标记。注意打或下传乘标记时还要将这个节点的加标记也一起乘上乘标记，因为(v[x]+tagpl[x])*tagmu[x]=v[x]*tagmu[x]+tagpl[x]*tagmu[x]。关于用什么堆，网上有人用斜堆结果T了，所以用复杂度稍微好一点的左偏树，只比斜堆多一个存节点最大深度的数组和多一句判断，也不难写。（斜堆也试了一下，也就慢了1s半，不知道是我斜堆(7s)写得好还是左偏树(5.5s)写得烂）。反思：脑子进水了，乘tagmu[x]写出乘x，wa了四次QAQ~

代码：

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 #define inc(i,j,k) for(int i=j;i<=k;i++)
 5 #define ll long long
 6 #define maxn 300010
 7 using namespace std;
 8
 9 inline ll read(){
10     char ch=getchar(); ll f=1,x=0;
11     while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1; ch=getchar();}
12     while(ch>=‘0‘&&ch<=‘9‘)x=x*10+ch-‘0‘,ch=getchar();
13     return f*x;
14 }
15 int n,m;
16 struct e{int t,n;}; e es[maxn]; int g[maxn],ess;
17 void pe(int f,int t){es[++ess]=(e){t,g[f]}; g[f]=ess;}
18 ll h[maxn],opt[maxn],p[maxn],ans[maxn],rt[maxn],die[maxn],d[maxn],st[maxn];
19 ll ch[maxn][2],v[maxn],tagmu[maxn],tagpl[maxn],dep[maxn];
20 void pushdown(int x){
21     int lc=ch[x][0],rc=ch[x][1];
22     if(tagmu[x]!=1||tagpl[x]!=0){
23         if(lc)v[lc]=v[lc]*tagmu[x]+tagpl[x],
24             tagmu[lc]*=tagmu[x],tagpl[lc]=tagpl[lc]*tagmu[x]+tagpl[x];
25         if(rc)v[rc]=v[rc]*tagmu[x]+tagpl[x],
26             tagmu[rc]*=tagmu[x],tagpl[rc]=tagpl[rc]*tagmu[x]+tagpl[x];
27         tagpl[x]=0; tagmu[x]=1;
28     }
29 }
30 int merge(int x,int y){
31     if(!x||!y)return x+y; if(v[y]<v[x])swap(x,y); pushdown(x);
32     ch[x][1]=merge(ch[x][1],y); if(dep[ch[x][0]]<dep[ch[x][1]])swap(ch[x][0],ch[x][1]);
33     dep[x]=dep[ch[x][0]]+1; return x;
34 }
35 int pop(int x){
36     pushdown(x); int y=merge(ch[x][0],ch[x][1]); ch[x][0]=ch[x][1]=0; return y;
37 }
38 int dfs(int x){
39     int y=rt[x];
40     for(int i=g[x];i;i=es[i].n)d[es[i].t]=d[x]+1,dfs(es[i].t),y=merge(y,rt[es[i].t]);
41     while(y&&v[y]<h[x])die[y]=x,y=pop(y),ans[x]++;
42     if(x!=1){
43         if(opt[x])v[y]*=p[x],tagmu[y]*=p[x],tagpl[y]*=p[x];else v[y]+=p[x],tagpl[y]+=p[x];
44     }
45     rt[x]=y; return y;
46 }
47 int main(){
48     n=read(); m=read(); inc(i,1,n)h[i]=read();
49     inc(i,2,n){int a=read(); pe(a,i); opt[i]=read(); p[i]=read();}
50     inc(i,1,m){
51         v[i]=read(); tagmu[i]=1; dep[i]=1; st[i]=read(); rt[st[i]]=merge(rt[st[i]],i);
52     }
53     d[1]=1; int x=dfs(1); inc(i,1,n)printf("%lld\n",ans[i]);
54     inc(i,1,m)printf("%lld\n",d[st[i]]-d[die[i]]);
55     return 0;
56 }

20160711