题意:
小铭铭最近获得了一副新的桌游,游戏中需要用 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。注意每个骑士是单独计算的。也就是说一个骑士攻击一座城池,不管结果如何,均不会影响其他骑士攻击这座城池的结果。
现在的问题是,对于每个城池,输出有多少个骑士在这里牺牲;对于每个骑士,输出他攻占的城池数量。
题解:
显然可并堆可以维护,从底往上合并,合并的时候打个标记即可。
code:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
struct node{
int y,next;
}a[300010];int len=0,last[300010];
struct trnode{
int lc,rc,d;
LL c,t1,t2;
}tr[300010];
void ins(int x,int y)
{
a[++len].y=y;
a[len].next=last[x];last[x]=len;
}
int n,m,op[300010],c[300010],root[300010],ans[300010],num[300010],dep[300010];
LL h[300010],v[300010];
void pushdown(int x)
{
LL t1=tr[x].t1,t2=tr[x].t2;tr[x].t1=1;tr[x].t2=0;
int lc=tr[x].lc,rc=tr[x].rc;
if(lc) tr[lc].c=tr[lc].c*t1+t2,tr[lc].t1*=t1,tr[lc].t2=tr[lc].t2*t1+t2;
if(rc) tr[rc].c=tr[rc].c*t1+t2,tr[rc].t1*=t1,tr[rc].t2=tr[rc].t2*t1+t2;
}
int merge(int x,int y)
{
if(!x||!y) return x+y;
pushdown(x);pushdown(y);
if(tr[x].c>tr[y].c) swap(x,y);
tr[x].rc=merge(tr[x].rc,y);
if(tr[tr[x].lc].d<tr[tr[x].rc].d) swap(tr[x].lc,tr[x].rc);
tr[x].d=tr[tr[x].rc].d+1;
return x;
}
void dfs(int x,int fa)
{
dep[x]=dep[fa]+1;
for(int i=last[x];i;i=a[i].next)
{
int y=a[i].y;dfs(y,x);
if(op[y]==0) tr[root[y]].c+=v[y],tr[root[y]].t2+=v[y];
else tr[root[y]].c*=v[y],tr[root[y]].t1*=v[y],tr[root[y]].t2*=v[y];
root[x]=merge(root[x],root[y]);
}
while(root[x]&&tr[root[x]].c<h[x]) ans[root[x]]=x,num[x]++,pushdown(root[x]),root[x]=merge(tr[root[x]].lc,tr[root[x]].rc);
}
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 fa;scanf("%d %d %lld",&fa,&op[i],&v[i]);
ins(fa,i);
}
for(int i=1;i<=m;i++)
{
LL s;scanf("%lld %d",&s,&c[i]);
tr[i].c=s;tr[i].t1=1;tr[i].t2=0;
root[c[i]]=merge(root[c[i]],i);
}
dfs(1,0);
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]]);
}