bzoj 3730: 震波 (点分树+线段树)

3730: 震波

Time Limit: 15 Sec  Memory Limit: 256 MB
Submit: 501  Solved: 108
[Submit][Status][Discuss]

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

Source

最近敲这种一言不合代码就150+的点分树题已经麻木了QAQ

对于每个重心我们开两颗线段树（动态开）：

一颗线段树中，对点分树中以u为根的子树中的节点v，我们以他距u的距离为下标，点权为值插入线段树

另一颗里，对点分治中以u为根的子树中的节点v，我们以他距u的父亲节点距离为下标，点权为值插入线段树

然后？你问我然后？

然后就修改暴力爬树高修改

然后就查询暴力爬树高查询，并消除上一层在当前层中重复的部分，统计答案就好

这种暴力爬树高，消除子树影响的方法其实很套路，建议先用点分树切掉bzoj3924或者看我这篇题解，然后搞这道题，具体过程依旧不细说，自己写的总是好，要攒人品。

代码：

#include<iostream>#include<algorithm>#include<cstdio>#include<cstdlib>#include<cmath>#include<cstring>#include<string>#include<climits>#include<queue>#include<stack>#include<map>#include<set>#define N 500100#define M 1000200#define lowbit(x) x&(-x)using namespace std;inline int read(){int x=0,f=1;char ch;while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;}int n,m,v[N],ans;int head[N],pos;struct edge{int to,next;}e[M];inline void add(int a,int b){pos++;e[pos].to=b,e[pos].next=head[a],head[a]=pos;}int dist[N],ff[N][20],dep[N];void dfs(int u,int fa){dep[u]=dep[fa]+1;ff[u][0]=fa;for(int i=head[u];i;i=e[i].next){int v=e[i].to;if(v==fa)continue;dist[v]=dist[u]+1;dfs(v,u);}}void make(){for(int k=1,j=2;j<=n;j<<=1,k++)for(int i=1;i<=n;i++)ff[i][k]=ff[ff[i][k-1]][k-1];}inline int lca(int x,int y){if(dep[x]<dep[y])swap(x,y);int tmp=dep[x]-dep[y];for(int k=0,j=1;j<=tmp;j<<=1,k++)if(tmp&j)x=ff[x][k];while(x!=y){int j=0;while(ff[x][j]!=ff[y][j])j++;if(j)j--;x=ff[x][j],y=ff[y][j];}return x;}inline int dis(int x,int y){int lc=lca(x,y);return dist[x]+dist[y]-2*dist[lc];}int root[N][2],sz;int s[N<<4],ls[N<<4],rs[N<<4];inline void insert(int &u,int l,int r,int x,int val){if(!u)u=++sz;s[u]+=val;if(l==r)return;int mid=(l+r)>>1;if(x<=mid)insert(ls[u],l,mid,x,val);else insert(rs[u],mid+1,r,x,val);}inline int query(int u,int l,int r,int x,int y){if(!u)return 0;if(x<=l&&y>=r)return s[u];int mid=(l+r)>>1;if(y<=mid)return query(ls[u],l,mid,x,y);else if(x>mid)return query(rs[u],mid+1,r,x,y);else return query(ls[u],l,mid,x,mid)+query(rs[u],mid+1,r,mid+1,y);}int sum,f[N],size[N],rt;bool vis[N];void find_root(int u,int fa){size[u]=1,f[u]=0;for(int i=head[u];i;i=e[i].next){int vv=e[i].to;if(vv==fa||vis[vv])continue;find_root(vv,u);size[u]+=size[vv];f[u]=max(f[u],size[vv]);}f[u]=max(f[u],sum-size[u]);if(f[u]<f[rt])rt=u;}int p[N],d[N];void get_dep(int tr,int u,int fa){insert(root[tr][0],0,n,d[u],v[u]);for(int i=head[u];i;i=e[i].next){int v=e[i].to;if(v==fa||vis[v])continue;d[v]=d[u]+1;get_dep(tr,v,u);}}void find_dep(int tr,int u,int fa){insert(root[tr][1],0,n,d[u],v[u]);for(int i=head[u];i;i=e[i].next){int v=e[i].to;if(v==fa||vis[v])continue;d[v]=d[u]+1;find_dep(tr,v,u);}}void work(int u){vis[u]=1;d[u]=0;get_dep(u,u,0);for(int i=head[u];i;i=e[i].next){int vv=e[i].to;if(vis[vv])continue;sum=size[vv],rt=0;find_root(vv,0);p[rt]=u;d[vv]=1;find_dep(rt,vv,u);work(rt);}}void updata(int u,int val){int va=query(root[u][0],0,n,0,0);int ct=val-va;insert(root[u][0],0,n,0,ct);for(int i=u;p[i];i=p[i]){int dt=dis(u,p[i]);insert(root[p[i]][0],0,n,dt,ct);insert(root[i][1],0,n,dt,ct);}}int ask(int u,int k){int j=k;int ret=query(root[u][0],0,n,0,k);for(int i=u;p[i];i=p[i]){int dt=dis(p[i],u);ret+=query(root[p[i]][0],0,n,0,k-dt);ret-=query(root[i][1],0,n,0,k-dt);}return ret;}int main(){//freopen("in.txt","r",stdin);//freopen("my.txt","w",stdout);n=read(),m=read();for(int i=1;i<=n;i++)v[i]=read();for(int i=1;i<n;i++){int a=read(),b=read();add(a,b);add(b,a);}sum=f[0]=n;rt=0;dfs(1,0);make();find_root(1,0);work(rt);for(int i=1;i<=m;i++){int f=read(),x=read(),y=read();x^=ans,y^=ans;if(f==0)printf("%d\n",ans=ask(x,y));else  updata(x,y);}}