[poj3140]Contestants Division树形dp

题意:切掉树上的某条边,使分开的两棵树上各点的权值和差值最小。

与hdu2196不同的是,此题是点权,其他无太大差别,注意数据范围。

先求出每个节点的子树权值和,然后自底向上dp即可。取$\min (abs(sum - 2*dp[u]))$

 #include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn=1e6+;
const ll inf=1e17+;
ll head[maxn],tot,n,m,sum;
struct edge{
ll to;
ll nxt;
ll w;
}e[maxn<<];
void add_edge(int u,int v,int w){
e[tot].to=v;
e[tot].w=w;
e[tot].nxt=head[u];
head[u]=tot++;
} ll dp[maxn<<],cnt[maxn<<]; void dfs(ll u,ll fa){
for(int i=head[u];i!=-;i=e[i].nxt){
int v=e[i].to;
if(v==fa) continue;
dfs(v,u);
cnt[u]+=1ll*cnt[v];
}
dp[u]=min(dp[u],sum-*cnt[u]>=?sum-*cnt[u]:*cnt[u]-sum);
} inline int read(){
char k=;char ls;ls=getchar();for(;ls<''||ls>'';k=ls,ls=getchar());
int x=;for(;ls>=''&&ls<='';ls=getchar())x=(x<<)+(x<<)+ls-'';
if(k=='-')x=-x;return x;
} int main(){
int k=;
while(scanf("%lld%lld",&n,&m)!=EOF&&(n||m)){
fill(dp+,dp+n+,inf);
memset(head,-,sizeof head);
tot=;
sum=;
int a,b;
for(int i=;i<=n;i++){
cnt[i]=read();
sum+=1ll*cnt[i];
}
for(int i=;i<m;i++){
a=read();
b=read();
add_edge(a,b,);
add_edge(b,a,);
}
printf("Case %d: ",++k);
dfs(,-);
ll ans=inf;
for(int i=;i<=n;i++){
ans=min(ans,dp[i]);
}
printf("%lld\n",ans);
}
return ;
}
上一篇:大数据之Oozie——源码分析(一)程序入口


下一篇:Linq操作DataTable