BZOJ 4372: 烁烁的游戏(动态点分治)

4372: 烁烁的游戏

Time Limit: 30 Sec  Memory Limit: 512 MB
Submit: 804  Solved: 288
[Submit][Status][Discuss]

Description

Q x：询问x的点权。
M x d w：将树上与节点x距离不超过d的节点的点权均加上w。

Output

思路：

#include<bits/stdc++.h>
using namespace std;
#define cherry main
const int maxn=1e5+10;
const int len=1e5;
int n,m;
struct node
{
int lson,rson,sum;
}tree1[maxn*210],tree2[maxn*210];
vector<int>G[maxn];
int f[maxn][20],dist[maxn][20],dep[maxn],c1[maxn],c2[maxn];
int root,now_size,son[maxn],T1[maxn],T2[maxn],tol1,tol2;
bool vis[maxn];
void get_root(int v,int fa,int SIZE)
{
int ma=0;
son[v]=1;
for(int i=0;i<G[v].size();i++)
{
int to=G[v][i];
if(to==fa||vis[to]) continue;
get_root(to,v,SIZE);
son[v]+=son[to];
ma=max(ma,son[to]);
}
ma=max(ma,SIZE-son[v]);
if(ma<now_size)
{
now_size=ma;
root=v;
}
}
int creat_node1()
{
tree1[++tol1].lson=tree1[tol1].rson=tree1[tol1].sum=0;
}
int creat_node2()
{
tree2[++tol2].lson=tree2[tol2].rson=tree2[tol2].sum=0;
}
void get_dep(int v,int fa,int father,int d)
{
for(int i=0;i<G[v].size();i++)
{
int to=G[v][i];
if(to==fa||vis[to]) continue;
f[to][++dep[to]]=father;
dist[to][dep[to]]=d;
get_dep(to,v,father,d+1);
}
}
void build(int v,int fa)
{
vis[v]=1;
T1[v]=creat_node1();
T2[v]=creat_node2();
get_dep(v,fa,v,1);
for(int i=0;i<G[v].size();i++)
{
int to=G[v][i];
if(to==fa||vis[to]) continue;
get_root(root=to,v,now_size=son[to]);
build(root,v);
}
}
void add2(int L,int C,int l,int r,int rt)
{
if(l==r)
{
tree2[rt].sum+=C;
return;
}
int mid=(l+r)/2;
if(L<=mid)
{
if(tree2[rt].lson==0)
tree2[rt].lson=creat_node2();
}
else
{
if(tree2[rt].rson==0)
tree2[rt].rson=creat_node2();
}
tree2[rt].sum=tree2[tree2[rt].lson].sum+tree2[tree2[rt].rson].sum;
}
void add1(int L,int C,int l,int r,int rt)
{
if(l==r)
{
tree1[rt].sum+=C;
return;
}
int mid=(l+r)/2;
if(L<=mid)
{
if(tree1[rt].lson==0)
tree1[rt].lson=creat_node1();
}
else
{
if(tree1[rt].rson==0)
tree1[rt].rson=creat_node1();
}
tree1[rt].sum=tree1[tree1[rt].lson].sum+tree1[tree1[rt].rson].sum;
}
void update(int x,int y,int z)
{
else c1[x]+=z;
if(dep[x]&&y>dist[x][dep[x]])
else if(dep[x]&&y==dist[x][dep[x]])
c2[x]+=z;
for(int i=dep[x];i>0;i--)
{
if(y>dist[x][i])
else if(y==dist[x][i])
c1[f[x][i]]+=z;
if(i>1&&y>dist[x][i-1])
else if(i>1&&y==dist[x][i-1])
c2[f[x][i]]+=z;
}
}
int sum1(int L,int R,int l,int r,int rt)
{
if(l>=L&&r<=R) return tree1[rt].sum;
int ans=0,mid=(l+r)/2;
if(L<=mid) ans+=sum1(L,R,l,mid,tree1[rt].lson);
if(R>mid) ans+=sum1(L,R,mid+1,r,tree1[rt].rson);
return ans;
}
int sum2(int L,int R,int l,int r,int rt)
{
if(l>=L&&r<=R) return tree2[rt].sum;
int ans=0,mid=(l+r)/2;
if(L<=mid) ans+=sum2(L,R,l,mid,tree2[rt].lson);
if(R>mid) ans+=sum2(L,R,mid+1,r,tree2[rt].rson);
return ans;
}
int query(int x)
{
int ans=sum1(1,len,1,len,T1[x])+c1[x];
for(int i=dep[x];i>0;i--)
{
ans+=sum1(dist[x][i],len,1,len,T1[f[x][i]]);
ans-=sum2(dist[x][i],len,1,len,T2[f[x][i+1]]);
}
return ans;
}
int cherry()
{
scanf("%d%d",&n,&m);
for(int i=1;i<n;i++)
{
int x,y;scanf("%d%d",&x,&y);
G[x].push_back(y);
G[y].push_back(x);
}
get_root(root=1,0,now_size=n);
build(root,0);
for(int i=1;i<=n;i++) f[i][dep[i]+1]=i;
while(m--)
{
char t[2];int x,y,z;
scanf("%s",&t);
if(t[0]=='Q')
{
scanf("%d",&x);
printf("%d\n",query(x));
}
else
{
scanf("%d%d%d",&x,&y,&z);
update(x,y,z);
}
}
return 0;
}


©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客