BZOJ4372: 烁烁的游戏

https://lydsy.com/JudgeOnline/problem.php?id=4372

分析:

• 不是很难想的一道题，用树状数组维护点分树上每一层分治中心的点分序即可。

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
#define N 100050
#define db(x) cerr<<#x<<" = "<<x<<endl
int n,m,head[N],to[N<<1],nxt[N<<1],cnt;
int siz[N],root,tot,fk[N],dep[N],dis[N][20],fa[N][20];
int used[N],bg[N][2],ed[N][2],satori,pos[N][20][2],c[N*40];
struct A {int x,d;A() {}A(int x_,int d_) {x=x_,d=d_;}bool operator < (const A &u) const {return d<u.d;}
}a[N*40];
inline void add(int u,int v) {to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt;
}
void gr(int x,int y) {int i;siz[x]=1; fk[x]=0;for(i=head[x];i;i=nxt[i]) if(to[i]!=y&&!used[to[i]]) {gr(to[i],x); siz[x]+=siz[to[i]];fk[x]=max(fk[x],siz[to[i]]);}fk[x]=max(fk[x],tot-fk[x]);if(fk[x]<fk[root]) root=x;
}
void gd(int x,int y,int rt,int d) {int i;a[++satori]=A(x,d);a[satori+tot]=A(x,dis[x][dep[x]]);fa[x][++dep[x]]=rt;dis[x][dep[x]]=d;for(i=head[x];i;i=nxt[i]) if(to[i]!=y&&!used[to[i]]) {gd(to[i],x,rt,d+1);}
}
void solve(int x) {used[x]=1; int i;int all=tot;bg[x][0]=satori+1; ed[x][0]=satori+all;bg[x][1]=satori+all+1; ed[x][1]=satori+all+all;gd(x,0,x,0);satori+=all;sort(a+bg[x][0],a+ed[x][0]+1);sort(a+bg[x][1],a+ed[x][1]+1);for(i=bg[x][0];i<=ed[x][0];i++) {pos[a[i].x][dep[a[i].x]][0]=i;}for(i=bg[x][1];i<=ed[x][1];i++) {pos[a[i].x][dep[a[i].x]][1]=i;}for(i=head[x];i;i=nxt[i]) if(!used[to[i]]) {tot=siz[to[i]]; if(tot>siz[x]) tot=all-siz[x];root=0; gr(to[i],x); solve(root);}
}
void fix(int x,int o,int k,int v) {if(!x) return ;int b=bg[x][o],e=ed[x][o],l=b,r=e+1;while(l<r) {int mid=(l+r)>>1;if(a[mid].d<=k) l=mid+1;else r=mid;}l--;int t=l-b+1;for(;t;t-=t&(-t)) c[t+b]+=v;
}
int inq(int x,int o,int p) {int b=bg[x][o],e=ed[x][o],re=0;int t=p-b+1,lim=e-b+1;for(;t<=lim;t+=t&(-t)) re+=c[t+b]; return re;
}
void update(int x,int k,int v) {int i;for(i=dep[x];i;i--) if(k>=dis[x][i]) {// db(fa[x][i]);fix(fa[x][i],0,k-dis[x][i],v);fix(fa[x][i+1],1,k-dis[x][i],-v);}
}
int query(int x) {int i,re=0;for(i=dep[x];i;i--) re+=inq(fa[x][i],0,pos[x][i][0])+inq(fa[x][i+1],1,pos[x][i+1][1]);return re;
}
char opt[10];
int main() {scanf("%d%d",&n,&m);int i,x,y;for(i=1;i<n;i++) {scanf("%d%d",&x,&y);add(x,y); add(y,x);}tot=n; fk[0]=1<<30; gr(1,0); solve(root);while(m--) {int k;scanf("%s%d",opt,&x);if(opt[0]=='M') {scanf("%d%d",&k,&y);update(x,k,y);}else {printf("%d\n",query(x));}}
}
/*
7 6
1 2
1 4
1 5
2 3
2 7
5 6
M 1 1 2
Q 5
M 2 2 3
Q 3
M 1 2 1
Q 2
*/