HDU5274 Dylans loves tree(树链剖分+异或)

传送门:http://acm.hdu.edu.cn/showproblem.php?pid=5274
题意很简单,输入以后,查询的时候
0 x y,是把x点改成y。
1 x y,是查询[x,y]中,数字出现的次数是否都是偶数,出现奇数的数字<=1。
都是偶数的话,输出-1,有奇数的话,输出奇数的权值。
思路:这题也是一个裸题,但是这题的线段树维护的比较巧妙,因为一个数异或自己偶数次就是0,所以求一个区间异或和就行了。但是有个地方要注意,就是权值为0的话,异或自己,奇数次也是0。所以可以把所有的权值+1,输出的时候再-1就好了。如果+1的话,在修改权值的时候也不要忘记+1。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <iostream>
#include <vector>
#include <map>
#include <queue>
#include <ctime>
using namespace std;
#pragma comment(linker, "/STACK:102400000,102400000")
typedef long long LL;
typedef pair<int,int> pii;
#define PB push_back
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define calm (l+r)>>1
const int INF=(int)1e9+7;

const int maxn=100010;
struct EE{
    int to,next;
    EE(){}
    EE(int to,int next):to(to),next(next){}
}edge[maxn*2];
int n,q,Ecnt,head[maxn],tot,val[maxn];
int top[maxn],fa[maxn],num[maxn],son[maxn],id[maxn],rev[maxn],deep[maxn];

inline void addedge(int a,int b){
    edge[Ecnt]=EE(b,head[a]);
    head[a]=Ecnt++;
}

void dfs1(int s,int pre,int d){
    fa[s]=pre;deep[s]=d;num[s]=1;son[s]=-1;
    for(int i=head[s];~i;i=edge[i].next){
        int t=edge[i].to;
        if(t==pre)continue;
        dfs1(t,s,d+1);num[s]+=num[t];
        if(son[s]==-1||num[t]>num[son[s]]){
            son[s]=t;
        }
    }
}
void dfs2(int s,int rt){
    top[s]=rt;id[s]=++tot;rev[tot]=s;
    if(son[s]==-1)return;
    dfs2(son[s],rt);
    for(int i=head[s];~i;i=edge[i].next){
        int t=edge[i].to;
        if(t==fa[s]||t==son[s])continue;
        dfs2(t,t);
    }
}
//SegmentTree
int sum[maxn<<2];
inline void pushup(int rt){
    sum[rt]=sum[rt<<1]^sum[rt<<1|1];
}
void build(int l,int r,int rt){
    if(l==r){
        sum[rt]=val[rev[l]];
        return;
    }
    int m=calm;
    build(lson);build(rson);
    pushup(rt);
}
int query(int L,int R,int l,int r,int rt){
    if(L<=l&&r<=R){
        return sum[rt];
    }
    int m=calm,ans=0;
    if(L<=m)ans^=query(L,R,lson);
    if(R>m)ans^=query(L,R,rson);
    return ans;
}
void update(int x,int v,int l,int r,int rt){
    if(l==r){
        sum[rt]=v;
        return;
    }
    int m=calm;
    if(x<=m)update(x,v,lson);
    else update(x,v,rson);
    pushup(rt);
}
int findsum(int x,int y){
    int f1=top[x],f2=top[y],ans=0;
    while(f1!=f2){
        if(deep[f1]<deep[f2]){
            swap(f1,f2);swap(x,y);
        }
        ans^=query(id[f1],id[x],1,n,1);
        x=fa[f1];f1=top[x];
    }
    if(deep[x]>deep[y])swap(x,y);
    ans^=query(id[x],id[y],1,n,1);
    return ans;
}
int main()
{
    //freopen("/home/xt/code/acm/input.txt","r",stdin);
    int T;scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&q);
        memset(head,-1,sizeof head);
        Ecnt=0;tot=0;
        for(int i=1;i<n;i++){
            int a,b;scanf("%d%d",&a,&b);
            addedge(a,b);addedge(b,a);
        }
        for(int i=1;i<=n;i++){
            scanf("%d",&val[i]);
            val[i]++;
        }
        dfs1(1,0,1);dfs2(1,1);
        build(1,n,1);
        while(q--){
            int op;scanf("%d",&op);
            int a,b;scanf("%d%d",&a,&b);
            if(op==0){
                update(id[a],b+1,1,n,1);//更新的也要+1
            }
            else{
                printf("%d\n",findsum(a,b)-1);//输出-1
            }
        }
    }
    //printf("[Run in %.1fs]\n",(double)clock()/CLOCKS_PER_SEC);
    return 0;
}

你可能感兴趣的:(HDU5274 Dylans loves tree(树链剖分+异或))