http://poj.org/problem?id=3140
题意:
要删一条边使得剩余的两个分支 价值差最小。
思路:
先从下往上更新子树的,再从上往下更新父亲节点,然后找最小
代码:
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
using namespace std;
#define ll long long
const int maxn=1e6+10;
const ll inf=1e15;
int n,m;
struct node
{
int to,next;
}edge[maxn*2];
int head[maxn],tot;
ll w[maxn];
void addedge(int u,int v)
{
edge[tot].to=v;
edge[tot].next=head[u];
head[u]=tot++;
}
void init()
{
memset(head,-1,sizeof(head));
tot=0;
}
ll up[maxn],down[maxn];
void dfs(int u,int pre)
{
down[u]=w[u];
for(int i=head[u];~i;i=edge[i].next)
{
int v=edge[i].to;
if(v==pre)continue;
dfs(v,u);
down[u]+=down[v];
}
}
void dfss(int u,int pre)
{
//up[u]=w[u];
for(int i=head[u];~i;i=edge[i].next)
{
int v=edge[i].to;
if(v==pre)continue;
up[v]+=up[u];
up[v]+=down[u]-down[v];
dfss(v,u);
}
}
ll mi;
int main()
{
int cs=0;
while(~scanf("%d%d",&n,&m)&&m+n)
{
init();
int u,v;
for(int i=1;i<=n;i++)
scanf("%lld",&w[i]);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&u,&v);
addedge(u,v);
addedge(v,u);
}
memset(up,0,sizeof(up));
dfs(1,-1);
dfss(1,-1);
mi=inf;
for(int i=1;i<=n;i++)
{
ll z=down[i]-up[i];
if(z<0)z=-z;
mi=min(mi,z);
}
printf("Case %d: %lld\n",++cs,mi);
}
return 0;
}