HDU 5274 Dylans loves tree 树链剖分(水)

HDU 5274
题意:n个节点的树,第i个节点权值为a[i].Q次操作.
操作1:修改a[x]的点权为y
操作2;询问[x,y]路径中是否有权值出现奇数次(保证出现奇数次的权值最多一个)
n,Q,a[i]<=1e5


修改点权 查询路径中是否有点权出现奇数次.

树链剖分后 维护区间的异或和即可. 注意权值可能为0 全部离散化,每个值+1,最后答案减去1即可.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=4e5+20;
int n,Q,cnt,sz;
int v[N],dep[N],size[N],head[N],fa[N];
int pos[N],bl[N];
struct edge{int to,next;}e[N];
struct node{int l,r,sum;}t[N<<2];
void add_edge(int u,int v){e[++cnt].to=v,e[cnt].next=head[u],head[u]=cnt;}

void init()
{
    memset(head,-1,sizeof(head));
    sz=cnt=0;//
    scanf("%d%d",&n,&Q);
    for(int i=1;i<=n-1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add_edge(x,y);
        add_edge(y,x);
    }
    for(int i=1;i<=n;i++)
        scanf("%d",&v[i]),v[i]++;
}
void dfs(int u)
{
    size[u]=1;
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        int v=e[i].to;
        if(v==fa[u])
            continue;
        dep[v]=dep[u]+1,fa[v]=u;
        dfs(v);
        size[u]+=size[v];
    }
}
void dfs(int x,int chain)
{
    int k=0;sz++;
    pos[x]=sz;//time;
    bl[x]=chain;
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int v=e[i].to;
        if(dep[v]>dep[x]&&size[v]>size[k])
            k=v;
    }
    if(k==0)
       return;
    dfs(k,chain);
    for(int i=head[x];i!=-1;i=e[i].next)
        if(dep[e[i].to]>dep[x]&&k!=e[i].to)
            dfs(e[i].to,e[i].to);
}
void build(int o,int l,int r)
{
    t[o].l=l,t[o].r=r,t[o].sum=0;
    if(l==r)
        return;
    int m=l+r>>1;
    build(o<<1,l,m);
    build(o<<1|1,m+1,r);
}
void push_up(int o)
{
    t[o].sum=t[o<<1].sum ^ t[o<<1|1].sum;
}
void change(int o,int x,int y)
{
    int l=t[o].l,r=t[o].r,mid=l+r>>1;
    if(l==r)
    {
        t[o].sum=y;
        return;
    }
    if(x<=mid)
        change(o<<1,x,y);
    else
        change(o<<1|1,x,y);
    push_up(o);
}
int querysum(int o,int ql,int qr)
{
    int l=t[o].l,r=t[o].r,m=l+r>>1;
    if(ql<=l&&qr>=r)
        return t[o].sum;
    int res=0;
    if(ql<=m)
        res^=querysum(o<<1,ql,qr);
    if(qr>m)
        res^=querysum(o<<1|1,ql,qr);
    return res;
}
int solvesum(int x,int y)
{
    int res=0;
    while(bl[x]!=bl[y])
    {
        if(dep[bl[x]]<dep[bl[y]])
            swap(x,y);
        res^=querysum(1,pos[bl[x]],pos[x]);
        x=fa[bl[x]];
    }
    if(pos[x]>pos[y])
        swap(x,y);
    res^=querysum(1,pos[x],pos[y]);
    return res;
}
void solve()
{
    build(1,1,n);
    for(int i=1;i<=n;i++)
        change(1,pos[i],v[i]);
    int op,x,y;
    while(Q--)
    {
        scanf("%d%d%d",&op,&x,&y);
        if(op==0)
            ++y,change(1,pos[x],y);
        else
        {
            int val=solvesum(x,y);
            printf("%d\n",val-1);
        }
    }
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        init();
        dfs(1);
        dfs(1,1);
        solve();
    }
    return 0;
}