poj 3140 树型DP

来源:互联网 发布:自动识别图片文字软件 编辑:程序博客网 时间:2024/05/28 23:22

题意:

一颗树,去掉某条边,使得剩下的两部分差值最小。问最小差值是多少

题解:

sum数组存储该节点及其子树的和。深搜一次算出sum,再枚举一次,得到最小差值。注意此题的long long

#include<stdio.h>#include<string.h>#include<algorithm>using namespace std;long long sum1,n,m,value[100010],list[100010],sum[100010],pos;struct P{long long u,v,next;}point[200010];void add(int son,int fa){point[pos].u=fa;point[pos].v=son;point[pos].next=list[fa];list[fa]=pos++;}void dfs(int son,int fa){long long now=list[son];sum[son]=value[son];while(now!=-1){if(point[now].v==fa){now=point[now].next;continue;}dfs(point[now].v,son);sum[son]+=sum[point[now].v];now=point[now].next;}return;}long long ab(long long a){return a>0?a:-a;}int main(){long long Cas=1;while(scanf("%lld%lld",&n,&m)&&(n&&m)){memset(list,-1,sizeof(list));pos=sum1=0;for(int i=1;i<=n;i++){scanf("%lld",&value[i]);sum1+=value[i];}for(long long i=1;i<=m;i++){int a,b;scanf("%d%d",&a,&b);add(a,b);add(b,a);}dfs(1,0);long long minn=sum1+1;for(int i=1;i<=n;i++){long long it=sum1-sum[i];if(ab(it-sum[i])<minn){minn=ab(it-sum[i]);}}printf("Case %lld: %lld\n",Cas++,minn);}return 0;}



原创粉丝点击