HDU 5274 Dylans loves tree(树链剖分)
生活随笔
收集整理的這篇文章主要介紹了
HDU 5274 Dylans loves tree(树链剖分)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
?
【題目鏈接】?http://acm.hdu.edu.cn/showproblem.php?pid=5274
?
【題目大意】
給出一棵樹,每個點有一個權值,權值可修改,且大于等于0,詢問鏈上出現次數為奇數的數,題目保證每次詢問的鏈上最多只有一個數出現次數為奇數。如果不存在這樣的數,就輸出-1。
?
【題解】
題目等價于求鏈上點的異或和,樹鏈剖分,線段樹維護區間異或和,然后鏈上查詢即可,注意到存在權值為0的特殊情況,所以我們將更新的數字都+1,在最后處理答案的時候-1即可。
?
【代碼】
#include <cstdio> #include <algorithm> #include <climits> #include <cstring> using namespace std; const int N=300010; int val[N],tot,op,x,d[N],num[N],ed=0,u,w,n,m,i,v[N],vis[N],f[N],g[N],nxt[N],size[N],son[N],st[N],en[N],dfn,top[N],t;char ch; void add(int x,int y){v[++ed]=y;nxt[ed]=g[x];g[x]=ed;} void dfs(int x){size[x]=1;for(int i=g[x];i;i=nxt[i])if(v[i]!=f[x]){f[v[i]]=x,d[v[i]]=d[x]+1;dfs(v[i]),size[x]+=size[v[i]];if(size[v[i]]>size[son[x]])son[x]=v[i];} } void dfs2(int x,int y){if(x==-1)return;st[x]=++dfn;top[x]=y;if(son[x])dfs2(son[x],y);for(int i=g[x];i;i=nxt[i])if(v[i]!=son[x]&&v[i]!=f[x])dfs2(v[i],v[i]);en[x]=dfn; } struct Node{int l,r;int sum;}T[N*4]; void build(int x,int l,int r){T[x].l=l;T[x].r=r;T[x].sum=0;if(l==r)return;int mid=(l+r)>>1;build(x<<1,l,mid);build((x<<1)|1,mid+1,r); } void up(int x){T[x].sum=T[x<<1].sum^T[x<<1|1].sum;} void update(int x,int k,int val){if(T[x].l==k&&T[x].r==k){T[x].sum^=val;return;}int mid=(T[x].l+T[x].r)>>1;if(k<=mid)update(x<<1,k,val);else update((x<<1)|1,k,val); up(x); } int sum(int x,int l,int r){if(l<=T[x].l&&T[x].r<=r)return T[x].sum;int mid=(T[x].l+T[x].r)>>1,tmp=0;if(l<=mid)tmp^=sum(x<<1,l,r);if(r>mid)tmp^=sum((x<<1)|1,l,r);return tmp; } int chain(int x,int y){int res=0;for(;top[x]!=top[y];x=f[top[x]]){if(d[top[x]]<d[top[y]]){int z=x;x=y;y=z;}res^=sum(1,st[top[x]],st[x]);}if(d[x]<d[y]){int z=x;x=y;y=z;}res^=sum(1,st[y],st[x]);return res; } void init(){ memset(g,dfn=ed=tot=0,sizeof(g));memset(v,0,sizeof(v));memset(nxt,0,sizeof(nxt));memset(son,-1,sizeof(son)); } int cas; int main(){scanf("%d",&cas);while(cas--){init();scanf("%d%d",&n,&m);for(int i=1;i<n;i++){scanf("%d%d",&u,&w);add(u,w); add(w,u); }dfs(1);dfs2(1,1);build(1,1,dfn);for(int i=1;i<=n;i++){scanf("%d",&val[i]); val[i]++;update(1,st[i],val[i]);}while(m--){scanf("%d%d%d",&op,&u,&w);if(op){printf("%d\n",chain(u,w)-1);}else{update(1,st[u],val[u]^(w+1));val[u]=w+1;}}}return 0; }
轉載于:https://www.cnblogs.com/forever97/p/hdu5274.html
總結
以上是生活随笔為你收集整理的HDU 5274 Dylans loves tree(树链剖分)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 信号、系统与滤波器设计(matlab)
- 下一篇: ES6入门之对象扩展