首页 > 代码库 > ZOJ 3684 Destroy 树的中心
ZOJ 3684 Destroy 树的中心
中心节点就是树的中心,2遍dfs求到树的直径。而中心一定在直径上,顺着直径找到中心就够了。
然后能够一遍树形DP找到最小值或者二分+推断是否訪问到叶子节点。
#include <iostream> #include<vector> #include<cstdio> #include<algorithm> using namespace std; struct node { int next; int power; int length; }t; vector<node> tree[10005]; int st,ed,maxs,n; void dfs1(int now,int fa,int sum) { if(sum>maxs) { maxs=sum; st=now; } for(int i=0;i<tree[now].size();i++) { int to=tree[now][i].next; int length=tree[now][i].length; if(to!=fa) { dfs1(to,now,sum+length); } } } struct node2 { int fa; int len; }tzf[10005]; int top,num; void dfs2(int now,int fa,int sum) { if(sum>maxs) { maxs=sum; ed=now; } for(int i=0;i<tree[now].size();i++) { int to=tree[now][i].next; int length=tree[now][i].length; if(to!=fa) { tzf[to].fa=now; tzf[to].len=length; dfs2(to,now,sum+length); } } } bool cal(int now,int fa,int lim) { if(tree[now].size()==1) { return false; } for(int i=0;i<tree[now].size();i++) { int to=tree[now][i].next; int power=tree[now][i].power; if(to!=fa&&power>lim) { if(!cal(to,now,lim)) return false; } } return true; } int main() { int maxx; int a,b,c,d; while(~scanf("%d",&n)) { maxx=0; for(int i=1;i<=n;i++) tree[i].clear(); for(int i=1;i<n;i++) { scanf("%d%d%d%d",&a,&b,&c,&d); maxx=max(maxx,d); t.length=c;; t.power=d; t.next=b; tree[a].push_back(t); t.next=a; tree[b].push_back(t); } maxs=0; dfs1(1,-1,0); maxs=0; tzf[st].fa=-1; dfs2(st,-1,0); int now=ed,mins=0x3f3f3f3f; int zx; int sum=0; while(now!=-1) { int k=max(sum,maxs-sum); if(k<mins) { mins=k; zx=now; } sum+=tzf[now].len; now=tzf[now].fa; } int l=0,r=maxx; int ans=0; while(l<=r) { int mid=(l+r)>>1; if(cal(zx,-1,mid)) { ans=mid; r=mid-1; } else { l=mid+1; } } printf("%d\n",ans); } return 0; } /* 7 1 2 8 2 1 3 2 2 3 6 4 0 2 4 3 0 2 5 10 0 5 7 12 0 */
ZOJ 3684 Destroy 树的中心
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。