首页 > 代码库 > 【poj3237】 Tree
【poj3237】 Tree
[http://poj.org/problem?id=3237] (题目链接)
树链剖分模板题,然而这150+行的程序我调了一天,历经艰辛,终于ac。。
题意:给出一个n个节点的带权树,要求维护操作:1.求出树上两点之间的边权的最大值;2.更改一条边上的权值;3.将树上两点之间的所有边权取各自的相反数。
solution
神奇的树链剖分+线段树维护查询和修改操作。
树链剖分时,我们将每条边的权值转换为除树根外每个节点上的权值(也就是对于每个节点与它父亲的边的权值转换到了自己的权值)。
之后就是标准的树链剖分后跑线段树了,那个全部取相反数的操作其实是一样的,树链剖分相关知识请见http://blog.sina.com.cn/s/blog_7a1746820100wp67.html
编程时请注意细节,邻接表写错了就悲剧了。。
为了避免各oier先WA后TLE在狂RE,附上造数据程序:
#include <ctime>#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>using namespace std;int main() { int i,j,k; freopen("aaa.in","r",stdin);freopen("aaa.in","w",stdout); puts("1"); srand((unsigned)time(NULL)); int n=rand()%1000+5; printf("%d\n",n); for(i=1;i<n;i++) printf("%d %d %d\n",i+1,rand()%i+1,rand()%14513546); for(i=1;i<n;i++) { int opt=rand()%3; if(opt==0) printf("CHANGE %d %d\n",rand()%(n-1)+1,rand()%42534567); else if(opt==1) { int a=0,b=0; while(a==b)a=rand()%n+1,b=rand()%n+1; printf("NEGATE %d %d\n",a,b); } else { int a=0,b=0; while(a==b)a=rand()%n+1,b=rand()%n+1; printf("QUERY %d %d\n",a,b); } } puts("DONE"); fclose(stdin);fclose(stdout); return 0;}
ac代码:
// poj3237#include<algorithm>#include<iostream>#include<cstring>#include<cstdlib>#include<cstdio>#include<cmath>#define MOD 1000000007#define inf 2147483640#define LL long long#define free(a) freopen(a".in","r",stdin);freopen(a".out","w",stdout);using namespace std;inline int getint() { int x=0,f=1;char ch=getchar(); while (ch>‘9‘ || ch<‘0‘) {if (ch==‘-‘) f=-1;ch=getchar();} while (ch>=‘0‘ && ch<=‘9‘) {x=x*10+ch-‘0‘;ch=getchar();} return x*f;}const int maxn=100010;struct edge {int to,next,w;}e[maxn<<2];struct tree {int l,r,tag,mn,mx;}tr[maxn<<2];int pos[maxn],deep[maxn],head[maxn],bin[20],fa[maxn][20],size[maxn],to[maxn],bl[maxn];int cnt,P,n;void insert(int u,int v,int w) { e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt;e[cnt].w=w; e[++cnt].to=u;e[cnt].next=head[v];head[v]=cnt;e[cnt].w=w;}void solve(int &x,int &y) { int t=x;x=-y;y=-t;}void update(int k){ tr[k].mn=min(tr[k<<1].mn,tr[k<<1|1].mn); tr[k].mx=max(tr[k<<1].mx,tr[k<<1|1].mx);}void pushdown(int k) { int l=tr[k].l,r=tr[k].r; if (l==r || !tr[k].tag) return; tr[k].tag=0; tr[k<<1].tag^=1,tr[k<<1|1].tag^=1; solve(tr[k<<1].mn,tr[k<<1].mx); solve(tr[k<<1|1].mn,tr[k<<1|1].mx);}void build(int k,int s,int t) { tr[k].l=s,tr[k].r=t,tr[k].tag=0,tr[k].mn=inf,tr[k].mx=-inf; if (s==t) return; int mid=(s+t)>>1; build(k<<1,s,mid); build(k<<1|1,mid+1,t);}void change(int k,int x,int val) { pushdown(k); int l=tr[k].l,r=tr[k].r,mid=(l+r)>>1; if (l==r) {tr[k].mn=tr[k].mx=val;return;} if (x<=mid) change(k<<1,x,val); else change(k<<1|1,x,val); update(k);}void rever(int k,int x,int y) { pushdown(k); int l=tr[k].l,r=tr[k].r,mid=(l+r)>>1; if (l==x && r==y) {solve(tr[k].mn,tr[k].mx);tr[k].tag=1;return;} if (y<=mid) rever(k<<1,x,y); else if (x>mid) rever(k<<1|1,x,y); else rever(k<<1,x,mid),rever(k<<1|1,mid+1,y); update(k);}int query(int k,int x,int y) { pushdown(k); int l=tr[k].l,r=tr[k].r,mid=(l+r)>>1; if (l==x && r==y) return tr[k].mx; if (y<=mid) return query(k<<1,x,y); else if (x>mid) return query(k<<1|1,x,y); else return max(query(k<<1,x,mid),query(k<<1|1,mid+1,y));}void dfs1(int x) { size[x]=1; for (int i=1;i<=13;i++) { if (bin[i]<=deep[x]) fa[x][i]=fa[fa[x][i-1]][i-1]; else break; } for (int i=head[x];i;i=e[i].next) if (e[i].to!=fa[x][0]) { deep[e[i].to]=deep[x]+1; fa[e[i].to][0]=x; dfs1(e[i].to); size[x]+=size[e[i].to]; }}void dfs2(int x,int chain) { /* if (x==22) { ++P; --P; } */ bl[x]=chain; pos[x]=++P; int k=0; for (int i=head[x];i;i=e[i].next) { if (e[i].to!=fa[x][0]) { if (size[e[i].to]>size[k]) k=e[i].to; } else { to[i>>1]=pos[x];//记录每个节点在线段树上的标号 change(1,pos[x],e[i].w);//将权值插入线段树 } } if (!k) return; dfs2(k,chain); for (int i=head[x];i;i=e[i].next) if (e[i].to!=fa[x][0] && e[i].to!=k) dfs2(e[i].to,e[i].to);}int lca(int x,int y) { if (deep[x]<deep[y]) swap(x,y); int t=deep[x]-deep[y]; for (int i=0;i<=13;i++) if (t&bin[i]) x=fa[x][i]; for (int i=13;i>=0;i--) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i]; if (x==y) return x; return fa[x][0];}int solvequery(int x,int f) { int mx=-inf; while (bl[x]!=bl[f]) { mx=max(mx,query(1,pos[bl[x]],pos[x])); x=fa[bl[x]][0]; } if (pos[f]+1<=pos[x]) mx=max(mx,query(1,pos[f]+1,pos[x])); return mx;}void solverever(int x,int f) { while (bl[x]!=bl[f]) { rever(1,pos[bl[x]],pos[x]); x=fa[bl[x]][0]; } if (pos[f]+1<=pos[x]) rever(1,pos[f]+1,pos[x]);}int main() { free("aaa"); int T=getint(); bin[0]=1;for (int i=1;i<15;i++) bin[i]=bin[i-1]<<1; while (T--) { P=0,cnt=1;//便于将边权转为点权 memset(head,0,sizeof(head)); memset(deep,0,sizeof(deep)); memset(fa,0,sizeof(fa)); n=getint(); for (int i=1;i<n;i++) { int u=getint(),v=getint(),w=getint(); insert(u,v,w); } build(1,1,n); dfs1(1); dfs2(1,1); char ch[10]; while (scanf("%s",ch+1)) { if (ch[1]==‘D‘) break; int x=getint(),y=getint(); if (ch[1]==‘Q‘) { int f=lca(x,y); printf("%d\n",max(solvequery(x,f),solvequery(y,f))); } if (ch[1]==‘C‘) change(1,to[x],y); if (ch[1]==‘N‘) { int f=lca(x,y); solverever(x,f);solverever(y,f); } } } fclose(stdin);fclose(stdout); return 0;}
【poj3237】 Tree
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。