【BZOJ3730】震波 动态树分治+线段树

时间:2023-03-09 03:00:51
【BZOJ3730】震波 动态树分治+线段树

【BZOJ3730】震波

Description

在一片土地上有N个城市,通过N-1条无向边互相连接,形成一棵树的结构,相邻两个城市的距离为1,其中第i个城市的价值为value[i]。
不幸的是,这片土地常常发生地震,并且随着时代的发展,城市的价值也往往会发生变动。
接下来你需要在线处理M次操作:
0 x k 表示发生了一次地震,震中城市为x,影响范围为k,所有与x距离不超过k的城市都将受到影响,该次地震造成的经济损失为所有受影响城市的价值和。
1 x y 表示第x个城市的价值变成了y。
为了体现程序的在线性,操作中的x、y、k都需要异或你程序上一次的输出来解密,如果之前没有输出,则默认上一次的输出为0。

Input

第一行包含两个正整数N和M。
第二行包含N个正整数,第i个数表示value[i]。
接下来N-1行,每行包含两个正整数u、v,表示u和v之间有一条无向边。
接下来M行,每行包含三个数,表示M次操作。

Output

包含若干行,对于每个询问输出一行一个正整数表示答案。

Sample Input

8 1
1 10 100 1000 10000 100000 1000000 10000000
1 2
1 3
2 4
2 5
3 6
3 7
3 8
0 3 1

Sample Output

11100101

HINT

1<=N,M<=100000
1<=u,v,x<=N
1<=value[i],y<=10000
0<=k<=N-1

题解:直接动态树分治,对于每一个分治中心,我们用维护一棵线段树维护它分治子树中的所有节点,下标为节点到分治中心的距离,权值为城市的权值。同时为了防重,还要维护一个从父亲那里减掉的版本。查询时,从当前点一直向父亲移动,每次在线段树中查一下就行了。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int maxn=100010;
struct node
{
int ls,rs,sum;
}s[maxn*200];
int n,m,cnt,tot,root,mn,ans;
int to[maxn<<1],next[maxn<<1],head[maxn],v[maxn],fa[maxn],siz[maxn],r1[maxn],r2[maxn],vis[maxn],Log[maxn<<1];
int md[20][maxn<<1],pos[maxn],dep[maxn];
void getrt(int x,int fa)
{
siz[x]=1;
int i,tmp=0;
for(i=head[x];i!=-1;i=next[i]) if(!vis[to[i]]&&to[i]!=fa)
getrt(to[i],x),siz[x]+=siz[to[i]],tmp=max(tmp,siz[to[i]]);
tmp=max(tmp,tot-siz[x]);
if(tmp<mn) root=x,mn=tmp;
}
void solve(int x)
{
vis[x]=1;
for(int i=head[x];i!=-1;i=next[i]) if(!vis[to[i]]) tot=siz[to[i]],mn=1<<30,getrt(to[i],x),fa[root]=x,solve(root);
}
void dfs(int x)
{
md[0][++pos[0]]=dep[x],pos[x]=pos[0];
for(int i=head[x];i!=-1;i=next[i]) if(!dep[to[i]]) dep[to[i]]=dep[x]+1,dfs(to[i]),md[0][++pos[0]]=dep[x];
}
inline void add(int a,int b)
{
to[cnt]=b,next[cnt]=head[a],head[a]=cnt++;
}
inline int getmin(int a,int b)
{
if(a>b) swap(a,b);
int k=Log[b-a+1];
return min(md[k][a],md[k][b-(1<<k)+1]);
}
inline int dis(int a,int b)
{
return dep[a]+dep[b]-2*getmin(pos[a],pos[b]);
}
void updata(int l,int r,int &x,int a,int b)
{
if(!x) x=++tot;
s[x].sum+=b;
if(l==r) return ;
int mid=(l+r)>>1;
if(a<=mid) updata(l,mid,s[x].ls,a,b);
else updata(mid+1,r,s[x].rs,a,b);
}
int query(int l,int r,int x,int a,int b)
{
if(!x||a>b) return 0;
if(a<=l&&r<=b) return s[x].sum;
int mid=(l+r)>>1;
if(b<=mid) return query(l,mid,s[x].ls,a,b);
if(a>mid) return query(mid+1,r,s[x].rs,a,b);
return query(l,mid,s[x].ls,a,b)+query(mid+1,r,s[x].rs,a,b);
}
inline char nc(){
static char buf[100000],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int rd(){
char ch=nc();int sum=0;
while(!(ch>='0'&&ch<='9'))ch=nc();
while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();
return sum;
}
int main()
{
//freopen("bz3730.in","r",stdin);
//freopen("bz3730.out","w",stdout);
n=rd(),m=rd();
int a,b,c;
register int i,j,x,y;
memset(head,-1,sizeof(head));
for(i=1;i<=n;i++) v[i]=rd();
for(i=1;i<n;i++) a=rd(),b=rd(),add(a,b),add(b,a);
dep[1]=1,dfs(1),tot=n,mn=1<<30,getrt(1,0),solve(root);
for(i=2;i<=pos[0];i++) Log[i]=Log[i>>1]+1;
for(j=1;(1<<j)<=pos[0];j++) for(i=1;i+(1<<j)-1<=pos[0];i++) md[j][i]=min(md[j-1][i],md[j-1][i+(1<<(j-1))]);
tot=0;
for(i=1;i<=n;i++)
{
x=i;
while(x)
{
y=fa[x],updata(0,n,r1[x],dis(x,i),v[i]);
if(y) updata(0,n,r2[x],dis(y,i),v[i]);
x=y;
}
}
for(i=1;i<=m;i++)
{
c=rd(),a=rd()^ans,b=rd()^ans;
if(!c)
{
ans=0,x=a;
while(x)
{
y=fa[x],ans+=query(0,n,r1[x],0,b-dis(x,a));
if(y) ans-=query(0,n,r2[x],0,b-dis(y,a));
x=y;
}
printf("%d\n",ans);
}
else
{
x=a;
while(x)
{
y=fa[x],updata(0,n,r1[x],dis(x,a),b-v[a]);
if(y) updata(0,n,r2[x],dis(y,a),b-v[a]);
x=y;
}
v[a]=b;
}
}
return 0;
}