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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章