当前位置: 代码迷 >> 综合 >> HDU 5274 Dylans loves tree (树链剖分)
  详细解决方案

HDU 5274 Dylans loves tree (树链剖分)

热度:75   发布时间:2023-12-17 03:30:48.0

题意:

一棵树,每个点有权,两种操作,一种单点修改,一种问x,y上的路有没有出现奇数次的点,并输出权值

思路:

因为异或了偶数次就还是0,所以树链剖分以后,每次询问的时候把所有的值异或起来就行了,因为数据还很小,所以我们单次直接更新到底就行了

错误及反思:

一开始没看到权值能为0,权值为0的时候我们手动加以下就可以了

代码:

#include<bits/stdc++.h>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define fi first
#define se second
using namespace std;
const int N =100100;int segtree[N*4];
pair<int,int> e[N*2];
int tot,tid,n,q;
int top[N],si[N],fa[N],fi[N],son[N],depth[N],id[N],f[N];
int val[N];void build(int l,int r,int rt){if(l==r){segtree[rt]=val[f[l]];return ;}int m=(l+r)/2;build(lson);build(rson);segtree[rt]=segtree[rt<<1]^segtree[rt<<1|1];
}
void addedge(int x,int y)
{e[tot].fi=y;e[tot].se=fi[x];fi[x]=tot++;e[tot].fi=x;e[tot].se=fi[y];fi[y]=tot++;
}
void dfs1(int now,int bef,int dep)
{fa[now]=bef;depth[now]=dep;si[now]=1;for(int i=fi[now];i!=-1;i=e[i].se)if(e[i].fi!=bef){dfs1(e[i].fi,now,dep+1);si[now]+=si[e[i].fi];if(son[now]==-1) son[now]=e[i].fi;else son[now]=si[e[i].fi]>si[son[now]]?e[i].fi:son[now];}
}
void dfs2(int now,int tp)
{top[now]=tp;id[now]=tid++;if(son[now]!=-1) dfs2(son[now],tp);for(int i=fi[now];i!=-1;i=e[i].se)if(e[i].fi!=fa[now]&&e[i].fi!=son[now])dfs2(e[i].fi,e[i].fi);
}
void init()
{memset(segtree,0,sizeof(segtree));tot=0; tid=1;memset(first,-1,sizeof(first));memset(son,-1,sizeof(son));
}
int query(int L,int R,int l,int r,int rt){if(L<=l&&R>=r){return  segtree[rt];}int an=0;int m=(l+r)/2;if(m>=L) an^=query(L,R,lson);if(m<R) an^=query(L,R,rson);return an;
}
int modify(int L,int R)
{int f1=top[L],f2=top[R];int ans=0;while(f1!=f2){if(depth[f1]<depth[f2]){swap(f1,f2);swap(L,R);}ans^=query(id[f1],id[L],1,n,1);L=fa[f1];f1=top[L];}if(depth[L]>depth[R]) swap(L,R);ans^=query(id[L],id[R],1,n,1);return ans;
}
void update(int p,int v,int l,int r,int rt){if(l==r&&l==p){segtree[rt]=v;return ;}int m=(l+r)/2;if(p<=m) update(p,v,lson);else update(p,v,rson);segtree[rt]=segtree[rt<<1]^segtree[rt<<1|1];
}
int main(){int T;scanf("%d",&T);while(T--){init();scanf("%d%d",&n,&q);for(int i=0,u,v;i<n-1;i++){scanf("%d%d",&u,&v);addedge(u,v);}for(int i=1;i<=n;i++){scanf("%d",&val[i]);val[i]++;}dfs1(1,1,1);dfs2(1,1);for(int i=1;i<=n;i++)f[id[i]]=i;build(1,n,1);while(q--){int op,x,y;scanf("%d%d%d",&op,&x,&y);if(op==1){int te=modify(x,y);if(te==0) printf("-1\n");else printf("%d\n",te-1);}else{update(id[x],y+1,1,n,1);}}}
}
  相关解决方案