3730: 震波

3730: 震波

Time Limit: 15 Sec  Memory Limit: 256 MB
Submit: 635  Solved: 151
[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

[Submit][Status][Discuss]
考虑实施维护任意一条路径的信息，对于原树，构建成点分树就是点分治的时候每次选出的重心，用重心链构成树，这样树高是O(logn)剩下维护信息就很直白了，每个点用一些动态长度的树状数组维护里面的信息不过代码细节挺多的= =挑了好久

#include<iostream>#include<cstring>#include<cstdio>#include<vector>#include<algorithm>using namespace std; const int maxn = 1E5 + 10;const int INF = ~0U>>1; int n,m,LastAns,rt,Max,O,cnt,val[maxn],siz[maxn],L[maxn],MaxD[maxn];bool Huge[maxn]; vector <int> v[maxn],G[maxn],h[maxn],Num[maxn],Dep[maxn],A[maxn],B[maxn]; int min(const int &x,const int &y) {return x < y ? x : y;} void Dfs1(int x,int tot,int fa){    int ma = 0; siz[x] = 1;    for (int i = 0; i < v[x].size(); i++)    {        int to = v[x][i];        if (to == fa || Huge[to]) continue;        Dfs1(to,tot,x); siz[x] += siz[to]; ma = max(ma,siz[to]);    }    ma = max(ma,tot - siz[x]);    if (ma < Max) Max = ma,O = x;} void Dfs2(int x,int fa){    for (int i = 0; i < v[x].size(); i++)    {        int to = v[x][i]; if (to == fa || Huge[to]) continue;        L[to] = MaxD[to] = L[x] + 1; Dfs2(to,x); MaxD[x] = max(MaxD[x],MaxD[to]);    }} void Dfs3(int x,int o,int y,int num,int t1,int t2,int fa){    siz[x] = 1; Num[x].push_back(num); Dep[x].push_back(L[x]);    for (int i = L[x]; i <= t1; i += i&-i) A[o][i] += val[x];    for (int i = L[x]; i <= t2; i += i&-i) B[y][i] += val[x];    for (int i = 0; i < v[x].size(); i++)    {        int to = v[x][i];        if (Huge[to] || to == fa) continue;        Dfs3(to,o,y,num,t1,t2,x); siz[x] += siz[to];    }} int Work(int x,int tot){    Max = INF; Dfs1(x,tot,0); int o = O,Now = 0;    Huge[o] = 1; L[o] = MaxD[o] = 0; Dfs2(o,0);    for (int i = 0; i <= MaxD[o]; i++) A[o].push_back(0);    for (int i = 0; i < v[o].size(); i++)    {        int to = v[o][i]; if (Huge[to]) continue; ++cnt;        for (int j = 0; j <= MaxD[to]; j++) B[cnt].push_back(0);        Dfs3(to,o,cnt,Now++,MaxD[o],MaxD[to],0); h[o].push_back(cnt);    }    for (int i = 0; i < v[o].size(); i++)    {        int to = v[o][i];        if (Huge[to]) continue;        int Nex = Work(to,siz[to]);        G[o].push_back(Nex);    }    return o;} int Query(int x,int y,int k,int now){    int ret = 0;    if (x == y)    {        for (int i = min(k,A[x].size() - 1); i > 0; i -= i&-i) ret += A[x][i];        return ret + val[x];    }    else    {        int Nex = h[x][Num[y][now]],st = k - Dep[y][now];        for (int i = min(st,A[x].size() - 1); i > 0; i -= i&-i) ret += A[x][i];        for (int i = min(st,B[Nex].size() - 1); i > 0; i -= i&-i) ret -= B[Nex][i];        Nex = G[x][Num[y][now]]; if (Dep[y][now] <= k) ret += val[x];        return ret + Query(Nex,y,k,now + 1);    }} void Modify(int x,int y,int va,int now){    if (x == y) {val[x] = va; return;}    int Nex = h[x][Num[y][now]],dep = Dep[y][now];    for (int j = dep; j < A[x].size(); j += j&-j) A[x][j] += (va - val[y]);    for (int j = dep; j < B[Nex].size(); j += j&-j) B[Nex][j] += (va - val[y]);    Nex = G[x][Num[y][now]];    Modify(Nex,y,va,now + 1);} int getint(){    char ch = getchar(); int ret = 0;    while (ch < '0' || '9' < ch) ch = getchar();    while ('0' <= ch && ch <= '9')        ret = ret*10 + ch - '0',ch = getchar();    return ret;} int main(){    #ifdef DMC        freopen("DMC.txt","r",stdin);    #endif         n = getint(); m = getint();    for (int i = 1; i <= n; i++) val[i] = getint();    for (int i = 1; i < n; i++)    {        int x = getint(),y = getint();        v[x].push_back(y); v[y].push_back(x);    }    rt = Work(1,n);         for (int I = 1; I <= m; I++)    {        int typ = getint(),x = getint(),y = getint();        x ^= LastAns; y ^= LastAns; if (typ) Modify(rt,x,y,0);        else printf("%d\n",LastAns = Query(rt,x,y,0));    }    return 0;}