#题意#
小铭铭最近获得了一副新的桌游,游戏中需要用 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 <= 300000; 1 <= fi<i; 1 <= ci <= n; -10^18 <= hi,vi,si <= 10^18;ai等于1或者2;当 ai =1 时,vi > 0;保证任何时候骑士战斗力值的绝对值不超过 10^18。
#分析#
一开始的想法是倍增,据说可以过但是好麻烦。
看了题解发现原来可以用数据结构来做,就是像线段树合并那样从底下往上做,期间打打标记啥的就好了。
左偏树平衡树什么的都很资瓷啊。
#代码#
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=300005;
int n,m,num[N],ans[N],dep[N],cnt,last[N],c[N],a[N],root[N];
LL h[N],v[N];
struct edge{int to,next;}e[N];
struct tree{int l,r,dis;LL val,tag1,tag2;}t[N];
void addedge(int u,int v)
{
e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
}
void pushdown(int d)
{
LL u=t[d].tag1,v=t[d].tag2;t[d].tag1=1;t[d].tag2=0;
if (t[d].l) t[t[d].l].val=t[t[d].l].val*u+v,t[t[d].l].tag1*=u,t[t[d].l].tag2=t[t[d].l].tag2*u+v;
if (t[d].r) t[t[d].r].val=t[t[d].r].val*u+v,t[t[d].r].tag1*=u,t[t[d].r].tag2=t[t[d].r].tag2*u+v;
}
int merge(int x,int y)
{
if (!x||!y) return x+y;
pushdown(x);pushdown(y);
if (t[x].val>t[y].val) swap(x,y);
t[x].r=merge(t[x].r,y);
if (t[t[x].l].dis<t[t[x].r].dis) swap(t[x].l,t[x].r);
t[x].dis=t[t[x].l].dis+1;
return x;
}
void get_dep(int x,int fa)
{
dep[x]=dep[fa]+1;
for (int i=last[x];i;i=e[i].next) get_dep(e[i].to,x);
}
void solve(int x)
{
for (int i=last[x];i;i=e[i].next)
{
solve(e[i].to);
if (a[e[i].to]==0) t[root[e[i].to]].val+=v[e[i].to],t[root[e[i].to]].tag2+=v[e[i].to];
else t[root[e[i].to]].val*=v[e[i].to],t[root[e[i].to]].tag1*=v[e[i].to],t[root[e[i].to]].tag2*=v[e[i].to];
root[x]=merge(root[x],root[e[i].to]);
}
while (root[x]&&t[root[x]].val<h[x]) ans[root[x]]=x,pushdown(root[x]),root[x]=merge(t[root[x]].l,t[root[x]].r),num[x]++;
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++) scanf("%lld",&h[i]);
for (int i=2;i<=n;i++)
{
int x;
scanf("%d%lld%lld",&x,&a[i],&v[i]);
addedge(x,i);
}
for (int i=1;i<=m;i++)
{
LL s;
scanf("%lld%d",&s,&c[i]);
t[i].val=s;t[i].tag1=1;
root[c[i]]=merge(root[c[i]],i);
}
get_dep(1,0);
solve(1);
for (int i=1;i<=n;i++) printf("%d\n",num[i]);
for (int i=1;i<=m;i++) printf("%d\n",dep[c[i]]-dep[ans[i]]);
return 0;
}