bzoj 4372 烁烁的游戏 —— 动态点分治+树状数组

时间:2023-03-10 08:54:04
bzoj 4372 烁烁的游戏 —— 动态点分治+树状数组

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

本以为和 bzoj3730 一样,可以直接双倍经验了;

但要注意一下,树状数组不能查询0位置,所以再开一个 w 数组记录;

论 if 和 continue 的不同...如果要用到两个值,不要判断第一个后就 continue ...

代码如下:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
int const xn=1e5+;
int n,hd[xn],ct,to[xn<<],nxt[xn<<],w[xn];
int siz[xn],fa[xn][],dis[xn][],dep[xn],mx,rt;
bool vis[xn];
vector<ll>t[xn],fx[xn];
int rd()
{
int ret=,f=; char ch=getchar();
while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}
while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
return f?ret:-ret;
}
int Max(int x,int y){return x>y?x:y;}
int Min(int x,int y){return x<y?x:y;}
void add(int x,int y){to[++ct]=y; nxt[ct]=hd[x]; hd[x]=ct;}
void getrt(int x,int ff,int sum)
{
siz[x]=; int nmx=;
for(int i=hd[x],u;i;i=nxt[i])
{
if((u=to[i])==ff||vis[u])continue;
getrt(u,x,sum); siz[x]+=siz[u];
nmx=Max(nmx,siz[u]);
}
nmx=Max(nmx,sum-siz[x]);
if(nmx<mx)mx=nmx,rt=x;
}
void build(int x,int ff,int d)
{
for(int i=hd[x],u;i;i=nxt[i])
{
if((u=to[i])==ff||vis[u])continue;
fa[u][++dep[u]]=rt; dis[u][dep[u]]=d;
build(u,x,d+);
}
}
void work(int x,int sum)
{
vis[x]=; build(x,,);
t[x].resize(sum+); fx[x].resize(sum+);
for(int i=hd[x],u;i;i=nxt[i])
{
if(vis[u=to[i]])continue;
int ns=(siz[u]>siz[x]?sum-siz[x]:siz[u]);
mx=xn; getrt(u,,ns); work(rt,ns);
}
}
char dc[];
void ins(int nw,int x,int v){w[nw]+=v; for(x=Min(x,t[nw].size()-);x;x-=(x&-x))t[nw][x]+=v;}///w[nw]
ll query(int nw,int x){/*if(x==0)return w[nw];*/ ll ret=; for(;x<t[nw].size()&&x;x+=(x&-x))ret+=t[nw][x]; return ret;}
void ins2(int nw,int x,int v){for(x=Min(x,fx[nw].size()-);x;x-=(x&-x))fx[nw][x]+=v;}
ll query2(int nw,int x){ll ret=; for(;x<fx[nw].size()&&x;x+=(x&-x))ret+=fx[nw][x]; return ret;}
ll ask(int x)
{
ll ret=;
for(int i=dep[x];i;i--)
ret+=query(fa[x][i],dis[x][i])-(i==?:query2(fa[x][i],dis[x][i-]));
return ret+w[x];
} void change(int x,int d,int val)
{
for(int i=dep[x];i;i--)
{
//if(d<dis[x][i])continue;
if(d>=dis[x][i])ins(fa[x][i],d-dis[x][i],val);
if(d<dis[x][i-]||i==)continue;
ins2(fa[x][i],d-dis[x][i-],val);//
}
}
int main()
{
n=rd(); int m=rd();
for(int i=,x,y;i<n;i++)x=rd(),y=rd(),add(x,y),add(y,x);
mx=xn; getrt(,,n); work(rt,n);
for(int i=;i<=n;i++)fa[i][++dep[i]]=i;
for(int i=,x,d,val;i<=m;i++)
{
scanf("%s",dc+); x=rd();
if(dc[]=='Q')printf("%lld\n",ask(x));
else d=rd(),val=rd(),change(x,d,val);
}
return ;
}