1. 程式人生 > >bzoj 4372 爍爍的遊戲——動態點分治+樹狀陣列

bzoj 4372 爍爍的遊戲——動態點分治+樹狀陣列

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

和 bzoj 3070 震波 是一個套路。注意區間修改的話,樹狀陣列不能表示 dis = 0 的位置,所以要手動改父親的點權陣列。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define ll long long
using namespace std;
const int N=1e5+5,K=20;
int n,w[N],hd[N],xnt,to[N<<1
],nxt[N<<1],siz[N],rt,mn; int pre[N][K],dep[N],dis[N][K],fs[N],gs[N]; vector<ll> f[N],g[N]; bool vis[N]; int rdn() { int ret=0;bool fx=1;char ch=getchar(); while(ch>'9'||ch<'0'){if(ch=='-')fx=0;ch=getchar();} while(ch>='0'&&ch<='9')ret=ret*10+ch-'0',ch=getchar(); return
fx?ret:-ret; } int Mx(int a,int b){return a>b?a:b;} int Mn(int a,int b){return a<b?a:b;} void add(int x,int y){to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;} void add(int cr,int x,int k){for(x=Mn(x,fs[cr]);x;x-=(x&-x))f[cr][x]+=k;} ll qry(int cr,int x){ll ret=0;for(;x&&x<=fs[cr];x+=(x&-x))ret+=f[cr][x];return
ret;} void addx(int cr,int x,int k){for(x=Mn(x,gs[cr]);x;x-=(x&-x))g[cr][x]+=k;} ll qryx(int cr,int x){ll ret=0;for(;x&&x<=gs[cr];x+=(x&-x))ret+=g[cr][x];return ret;} void getrt(int cr,int fa,int s) { siz[cr]=1;int mx=0; for(int i=hd[cr],v;i;i=nxt[i]) if(!vis[v=to[i]]&&v!=fa) { getrt(v,cr,s);siz[cr]+=siz[v];mx=Mx(mx,siz[v]); } mx=Mx(mx,s-siz[cr]); if(mx<mn)mn=mx,rt=cr; } void dfs(int cr,int fa,int ds) { pre[cr][++dep[cr]]=rt;dis[cr][dep[cr]]=ds; for(int i=hd[cr],v;i;i=nxt[i]) if(!vis[v=to[i]]&&v!=fa)dfs(v,cr,ds+1); } void init(int cr,int s) { vis[cr]=1;fs[cr]=mn;gs[cr]=(mn<<1)+1; f[cr].resize(fs[cr]+1);g[cr].resize(gs[cr]+1); dfs(cr,0,0); for(int i=hd[cr],v;i;i=nxt[i]) if(!vis[v=to[i]]) { int ts=(siz[v]<siz[cr]?siz[v]:s-siz[cr]); mn=N;getrt(v,cr,ts);init(rt,ts); } } void mdfy(int cr,int d,int k) { add(cr,d,k); w[cr]+=k; for(int i=dep[cr]-1;i;i--) { if(dis[cr][i]>d)continue; add(pre[cr][i],d-dis[cr][i],k); w[pre[cr][i]]+=k;///// addx(pre[cr][i+1],d-dis[cr][i],k); } } ll query(int cr) { ll ret=w[cr]; for(int i=dep[cr]-1;i;i--) { ret+=qry(pre[cr][i],dis[cr][i]); ret-=qryx(pre[cr][i+1],dis[cr][i]); } return ret; } int main() { n=rdn(); int Q=rdn(); for(int i=1,u,v;i<n;i++) u=rdn(),v=rdn(),add(u,v),add(v,u); mn=N;getrt(1,0,n);init(rt,n); int x,d,k;char ch[5]; while(Q--) { scanf("%s",ch);x=rdn(); if(ch[0]=='Q')printf("%lld\n",query(x)); else { d=rdn(); k=rdn(); mdfy(x,d,k); } } return 0; }