bzoj2819 DFS序 + LCA + 线段树

https://www.lydsy.com/JudgeOnline/problem.php?id=2819

题意：树上单点修改及区间异或和查询。

思维难度不高，但是题比较硬核。

整体思路是维护每一个结点到根节点的距离。查询u，v树链上的异或和就是query(v) ^ query(u) ^ a[lca(u,v)],所以就要想办法维护树上的结点到根节点的异或和。

网上的题解大多是选择直接维护答案，修改的时候修改整颗子树的答案，用线段树或者树状数组区间异或一下l到r内的答案就可以。

我一开始没有想到直接上lca直接维护根节点的距离，所以在dfs序上见了一颗线段树，查询的时候query 1到u第一次出现地方的异或和，区间的时候常规单点修改也可过。

#include <map>
#include <set>
#include <ctime>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <sstream>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
inline int read(){int now=;register char c=getchar();for(;!isdigit(c);c=getchar());
for(;isdigit(c);now=now*+c-'',c=getchar());return now;}
#define For(i, x, y) for(int i=x;i<=y;i++)
#define _For(i, x, y) for(int i=x;i>=y;i--)
#define Mem(f, x) memset(f,x,sizeof(f))
#define Sca(x) scanf("%d", &x)
#define Sca2(x,y) scanf("%d%d",&x,&y)
#define Sca3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define Scl(x) scanf("%lld",&x);
#define Pri(x) printf("%d\n", x)
#define Prl(x) printf("%lld\n",x);
#define CLR(u) for(int i=0;i<=N;i++)u[i].clear();
#define LL long long
#define ULL unsigned long long
#define mp make_pair
#define PII pair<int,int>
#define PIL pair<int,long long>
#define PLL pair<long long,long long>
#define pb push_back
#define fi first
#define se second
typedef vector<int> VI;
const double eps = 1e-;
const int maxn = 5e5 + ;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + ;
int N,M,K;
LL a[maxn];
struct Edge{
    int next,to;
}edge[maxn * ];
int head[maxn],tot;
void init(){
    for(int i = ; i <= N ; i ++) head[i] = -;
    tot = ;
}
void add(int u,int v){
    edge[tot].to = v;
    edge[tot].next = head[u];
    head[u] = tot++;
}
const int SP = ;
int pa[maxn][SP],dep[maxn];
//求dfs序
int num;
int dfs_index[maxn * ];
PII pos[maxn];
void dfs(int t,int la){
    pa[t][] = la;
    dep[t] = dep[la] + ;
    for(int i = ; i <= SP - ; i ++) pa[t][i] = pa[pa[t][i - ]][i - ];
    pos[t].fi = ++num;
    dfs_index[num] = t;
    for(int i = head[t]; ~i ; i = edge[i].next){
        int v = edge[i].to;
        if(v == la) continue;
        dfs(v,t);
    }
    pos[t].se = ++num;
    dfs_index[num] = t;
}
int lca(int u,int v){
    if(dep[u] < dep[v]) swap(u,v);
    int t = dep[u] - dep[v];
    for(int i =  ; i <= SP - ; i ++){
        if(t & ( << i)) u = pa[u][i];
    }
    for(int i = SP - ; i >=  ; i --){
        int uu = pa[u][i],vv = pa[v][i];
        if(uu != vv){
            u = uu;
            v = vv;
        }
    }
    return u == v? u :pa[u][];
}
//线段树
struct Tree{
    LL sum;
}tree[maxn << ];
void Pushup(int t){
    tree[t].sum = tree[t << ].sum ^ tree[t <<  | ].sum;
}
void Build(int t,int l,int r){
    if(l == r){
        tree[t].sum = a[dfs_index[l]];
        return;
    }
    int m = (l + r) >> ;
    Build(t << ,l,m); Build(t <<  | ,m + ,r);
    Pushup(t);
}
void update(int t,int p,int x,int L,int R){
    if(L == R){
        tree[t].sum = x;
        return;
    }
    int m = (L + R) >> ;
    if(p <= m) update(t << ,p,x,L,m);
    else update(t <<  | ,p,x,m + ,R);
    Pushup(t);
}
LL query(int t,int l,int r,int L,int R){
    if(l <= L && R <= r){
        return tree[t].sum;
    }
    int m = (L + R) >> ;
    if(r <= m) return query(t << ,l,r,L,m);
    else if(l > m) return query(t <<  | ,l,r,m + ,R);
    else return query(t << ,l,m,L,m) ^ query(t <<  | ,m + ,r,m + ,R);
}

LL query(int x){
    return query(,,pos[x].fi,, * N);
}
int main(){
    Sca(N); init();
    For(i,,N) Scl(a[i]);
    For(i,,N - ){
        int u,v; Sca2(u,v);
        add(u,v); add(v,u);
    }
    dfs(,);
    Build(,, * N);
    Sca(K);
    while(K--){
        char op[];
        int u,v;
        scanf("%s%d%d",op,&u,&v);
        if(op[] == 'Q'){
            if(query(u) ^ query(v) ^ a[lca(u,v)]) puts("Yes");
            else puts("No");
        }else{
            update(,pos[u].fi,v,, * N);
            update(,pos[u].se,v,, * N);
            a[u] = v;
        }
    }
    return ;
}