[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 攻占的城池数量。
放三个懒标记
分别维护战斗力累乘(lazyc),累加(lazy),以及攻略城市的累加(addv),在查询时释放
是不是很像线段树?
QwQ
树形遍历
#include<bits/stdc++.h> #define re return #define R register #define ll long long #define inc(i,l,r) for(register int i=l;i<=r;++i) using namespace std; const int maxn=300005; char buf[1<<21],*p1,*p2; inline int gc(){re p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;} template<typename T>inline void rd(T&x) { char c;bool f=0; while((c=getchar())<'0'||c>'9')if(c=='-')f=1; x=c^48; while((c=getchar())>='0'&&c<='9')x=x*10+(c^48); if(f)x=-x; } ll n,m,hd[maxn]; ll ls[maxn],rs[maxn],val[maxn],dis[maxn]; //左偏树日常标记 ll pro[maxn],a[maxn],v[maxn],rt[maxn],size[maxn]; //pro城市生命值 ,rt能够攻略城市的骑士所在左偏树,size牺牲在这座城市的人数 ll cnt[maxn],lazyc[maxn],lazy[maxn],addv[maxn]; //cnt骑士攻占的城市数 struct node{ int to,nt; }e[maxn]; inline void pushup(int x) { if(addv[x]) { cnt[ls[x]]+=addv[x];cnt[rs[x]]+=addv[x]; addv[ls[x]]+=addv[x];addv[rs[x]]+=addv[x]; addv[x]=0; } if(lazyc[x]) { val[ls[x]]*=lazyc[x];val[rs[x]]*=lazyc[x]; lazy[ls[x]]*=lazyc[x];lazy[rs[x]]*=lazyc[x]; lazyc[ls[x]]*=lazyc[x];lazyc[rs[x]]*=lazyc[x]; lazyc[x]=1; } if(lazy[x]) { val[ls[x]]+=lazy[x];val[rs[x]]+=lazy[x]; lazy[ls[x]]+=lazy[x];lazy[rs[x]]+=lazy[x]; lazy[x]=0; } } inline int merge(int x,int y) { if(!x||(!y))re x+y; if(val[x]>val[y]) x^=y^=x^=y; pushup(x); //查询时释放,以防下层merge时val[rs[x]]出锅 rs[x]=merge(rs[x],y); if(dis[rs[x]]>dis[ls[x]])swap(ls[x],rs[x]); dis[x]=dis[rs[x]]+1; re x; } inline void dfs(int x) { for(int i=hd[x];i;i=e[i].nt) { R int vv=e[i].to; dfs(vv); rt[x]=merge(rt[x],rt[vv]); while(rt[x]&&val[rt[x]]<pro[x]) { pushup(rt[x]); //上一次merge有一子树为空时,另一子树的懒标记未曾释放 rt[x]=merge(ls[rt[x]],rs[rt[x]]); //踢出<城市生命值 ++size[x]; } } ++addv[rt[x]];++cnt[rt[x]]; if(a[x]==0){ lazy[rt[x]]+=v[x]; val[rt[x]]+=v[x]; } else { lazyc[rt[x]]*=v[x]; lazy[rt[x]]*=v[x]; val[rt[x]]*=v[x]; } } inline void dfsz(int x) { pushup(x); if(ls[x])dfsz(ls[x]); if(rs[x])dfsz(rs[x]); } int main() { // freopen("in.txt","r",stdin); int x; rd(n),rd(m); inc(i,1,n)rd(pro[i]); inc(i,2,n) { rd(x); e[i]=(node){i,hd[x]};hd[x]=i; rd(a[i]),rd(v[i]); } inc(i,1,m) { rd(val[i]); rd(x); lazyc[i]=1; if(val[i]<pro[x])++size[x]; else rt[x]=merge(i,rt[x]); } dfs(1); inc(i,1,n) printf("%d\n",size[i]); dfsz(rt[1]);
//树形遍历一遍,释放标记 inc(i,1,m) printf("%d\n",cnt[i]); re 0; }