[JLOI2015]城池攻占
题目描述
小铭铭最近获得了一副新的桌游,游戏中需要用 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。注意每个骑士是单独计算的。也就是说一个骑士攻击一座城池,不管结果如何,均不会影响其他骑士攻击这座城池的结果。
现在的问题是,对于每个城池,输出有多少个骑士在这里牺牲;对于每个骑士,输出他攻占的城池数量。
输入输出格式
输入格式:第 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,分别表示初始战斗力和第一个攻击的城池。
输出格式:输出 n + m 行,每行包含一个非负整数。其中前 n 行分别表示在城池 1 到 n 牺牲的骑士数量,后 m 行分别表示骑士 1 到 m 攻占的城池数量。
输入输出样例
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
2 2 0 0 0 1 1 3 1 1
说明
对于 100% 的数据,1 <= n;m <= 300000; 1 <= fi<i; 1 <= ci <= n; -10^18 <= hi,vi,si <= 10^18;ai等于1或者2;当 ai =1 时,vi > 0;保证任何时候骑士战斗力值的绝对值不超过 10^18。
每一个点维护一个小根堆
从叶子节点向上处理,把小于h[i]的pop掉,修改后和父亲合并
至于修改,给左偏堆打上延迟标记
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<cmath> 6 using namespace std; 7 typedef long long lol; 8 struct Node 9 { 10 int next,to; 11 }edge[300001]; 12 struct ZYYS 13 { 14 int dis; 15 lol v,add,mul; 16 int l,r; 17 }t[300001]; 18 int head[300001],num,rt[300001]; 19 lol h[300001],die[300001]; 20 int death[300001],a[300001],n,m; 21 lol v[300001]; 22 int fa[300001],dep[300001],c[300001]; 23 void add(int u,int v) 24 { 25 num++; 26 edge[num].next=head[u]; 27 head[u]=num; 28 edge[num].to=v; 29 } 30 void update(int x,lol add,lol mul) 31 { 32 if (!x) return; 33 if (add==0&&mul==1) return; 34 t[x].v=t[x].v*mul+add; 35 t[x].add=mul*t[x].add+add; 36 t[x].mul*=mul; 37 } 38 void pushdown(int x) 39 { 40 update(t[x].l,t[x].add,t[x].mul); 41 update(t[x].r,t[x].add,t[x].mul); 42 t[x].add=0;t[x].mul=1; 43 } 44 int merge(int a,int b) 45 { 46 if (!a||!b) return a+b; 47 if (t[a].v>t[b].v) swap(a,b); 48 pushdown(a);pushdown(b); 49 t[a].r=merge(t[a].r,b); 50 if (t[t[a].l].dis<t[t[a].r].dis) swap(t[a].l,t[a].r); 51 t[a].dis=t[t[a].r].dis+1; 52 return a; 53 } 54 void dfs(int x) 55 {int i; 56 for (i=head[x];i;i=edge[i].next) 57 { 58 int v=edge[i].to; 59 dep[v]=dep[x]+1; 60 dfs(v); 61 rt[x]=merge(rt[x],rt[v]); 62 } 63 while (rt[x]&&t[rt[x]].v<h[x]) 64 { 65 death[x]++; 66 pushdown(rt[x]); 67 die[rt[x]]=x; 68 rt[x]=merge(t[rt[x]].l,t[rt[x]].r); 69 } 70 if (a[x]) update(rt[x],0,v[x]); 71 else update(rt[x],v[x],1); 72 } 73 int main() 74 {int i; 75 cin>>n>>m; 76 for (i=1;i<=n;i++) 77 scanf("%lld",&h[i]); 78 for (i=2;i<=n;i++) 79 { 80 scanf("%d%d%lld",&fa[i],&a[i],&v[i]); 81 add(fa[i],i); 82 } 83 for (i=1;i<=m;i++) 84 { 85 scanf("%lld%d",&t[i].v,&c[i]); 86 t[i].mul=1;t[i].add=0; 87 rt[c[i]]=merge(rt[c[i]],i); 88 } 89 dep[1]=1; 90 dfs(1); 91 for (i=1;i<=n;i++) 92 printf("%d\n",death[i]); 93 for (i=1;i<=m;i++) 94 printf("%d\n",dep[c[i]]-dep[die[i]]); 95 }