首页 > 代码库 > BZOJ 3531 SDOI2014 旅行 树链剖分
BZOJ 3531 SDOI2014 旅行 树链剖分
题目大意:给定一棵树,每个点有一个权值和一个颜色,多次改变一些点的权值和颜色,多次求一条路径上与起点和终点颜色相同的点的权值和以及权值最大值
每种颜色开一个线段树 动态开节点 每个点只建一条链 这样空间复杂度是O(nlogn)的
然后就正常树链剖分就行了
#include <queue> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define M 100100 using namespace std; struct abcd{ int to,next; }table[M<<1]; struct Segtree{ Segtree *ls,*rs; int sum,max_num; Segtree():ls(0x0),rs(0x0),sum(0),max_num(0){} }*tree[M]; int head[M],tot; int n,m; int w[M],c[M]; int fa[M],son[M],dpt[M],size[M],pos[M],top[M],cnt; void Update(Segtree* &p,int x,int y,int z,int v) { int mid=x+y>>1; if(!p) p=new Segtree; if(x==y) { p->sum=p->max_num=v; return ; } if(z<=mid) Update(p->ls,x,mid,z,v); else Update(p->rs,mid+1,y,z,v); p->sum=(p->ls?p->ls->sum:0)+(p->rs?p->rs->sum:0); p->max_num=max( (p->ls?p->ls->max_num:0) , (p->rs?p->rs->max_num:0) ); } int Get_Sum(Segtree *p,int x,int y,int l,int r) { int mid=x+y>>1; if(!p) return 0; if(x==l&&y==r) return p->sum; if(r<=mid) return Get_Sum(p->ls,x,mid,l,r); if(l>mid) return Get_Sum(p->rs,mid+1,y,l,r); return Get_Sum(p->ls,x,mid,l,mid) + Get_Sum(p->rs,mid+1,y,mid+1,r); } int Get_Max(Segtree *p,int x,int y,int l,int r) { int mid=x+y>>1; if(!p) return 0; if(x==l&&y==r) return p->max_num; if(r<=mid) return Get_Max(p->ls,x,mid,l,r); if(l>mid) return Get_Max(p->rs,mid+1,y,l,r); return max( Get_Max(p->ls,x,mid,l,mid) , Get_Max(p->rs,mid+1,y,mid+1,r) ); } void Add(int x,int y) { table[++tot].to=y; table[tot].next=head[x]; head[x]=tot; } void DFS1(int x) { int i; dpt[x]=dpt[fa[x]]+1; size[x]=1; for(i=head[x];i;i=table[i].next) { if(table[i].to==fa[x]) continue; fa[table[i].to]=x; DFS1(table[i].to); size[x]+=size[table[i].to]; if(size[table[i].to]>size[son[x]]) son[x]=table[i].to; } } void DFS2(int x) { int i; pos[x]=++cnt; if(son[fa[x]]==x) top[x]=top[fa[x]]; else top[x]=x; Update(tree[c[x]],1,n,pos[x],w[x]); if(son[x]) DFS2(son[x]); for(i=head[x];i;i=table[i].next) { if(table[i].to==fa[x]||table[i].to==son[x]) continue; DFS2(table[i].to); } } int Query_Sum(int x,int y,int _c) { int fx=top[x],fy=top[y],re=0; while(fx!=fy) { if(dpt[fx]<dpt[fy]) swap(x,y),swap(fx,fy); re+=Get_Sum(tree[_c],1,n,pos[fx],pos[x]); x=fa[fx];fx=top[x]; } if(dpt[x]<dpt[y]) swap(x,y); re+=Get_Sum(tree[_c],1,n,pos[y],pos[x]); return re; } int Query_Max(int x,int y,int _c) { int fx=top[x],fy=top[y],re=0; while(fx!=fy) { if(dpt[fx]<dpt[fy]) swap(x,y),swap(fx,fy); re=max(re,Get_Max(tree[_c],1,n,pos[fx],pos[x])); x=fa[fx];fx=top[x]; } if(dpt[x]<dpt[y]) swap(x,y); re=max(re,Get_Max(tree[_c],1,n,pos[y],pos[x])); return re; } int main() { int i,x,y; char p[10]; cin>>n>>m; for(i=1;i<=n;i++) scanf("%d%d",&w[i],&c[i]); for(i=1;i<n;i++) scanf("%d%d",&x,&y),Add(x,y),Add(y,x); DFS1(1);DFS2(1); for(i=1;i<=m;i++) { scanf("%s%d%d",p,&x,&y); switch(p[1]) { case 'C': Update(tree[c[x]],1,n,pos[x],0); Update(tree[c[x]=y],1,n,pos[x],w[x]); break; case 'W': Update(tree[c[x]],1,n,pos[x],w[x]=y); break; case 'S': printf("%d\n", Query_Sum(x,y,c[x]) ); break; case 'M': printf("%d\n", Query_Max(x,y,c[x]) ); break; } } }
BZOJ 3531 SDOI2014 旅行 树链剖分
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。