POJ 3140 Contestants Division 树形dp

题意:

一个带权的树删去一条边使得分成的两部分权值和的差最小

分析:

dfs暴力枚举每一条边就好了

tips:inf 如果设为0x3f3f3f3f3f 会错 因为答案可能很大

ACcode:

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <vector>
#define maxn 100005
#define inf (((unsigned long long)(1)<<63)-1)
#define ll long long
using namespace std;
int n,m,tmp;
struct N{
    int to,next;
}my[maxn<<1];
int num[maxn];
ll sum,ans,total,dp[maxn];
int head[maxn],tot,cnt=1;
void init(){
    memset(head,-1,sizeof(head));
    total=sum=tot=0;
}
void add(int u,int v){
    my[tot].to=v;
    my[tot].next=head[u];
    head[u]=tot++;
    ans=inf;
}
void dfs(int u,int fa){
    for(int i=head[u];i!=-1;i=my[i].next){
        int v=my[i].to;
        if(fa==v)continue;
        dfs(v,u);
        ll tmp=total-2*dp[v];
        if(tmp<0)tmp=-tmp;
        ans=min(ans,tmp);
        dp[u]+=dp[v];
    }
}
int main(){
    while(scanf("%d%d",&n,&m)&&n+m){
        init();
        for(int i=1;i<=n;++i){
            scanf("%d",&num[i]);
            total+=num[i];
            dp[i]=num[i];
        }
        for(int i=1;i<n;++i){
            int u,v;
            scanf("%d%d",&u,&v);
            add(u,v);
            add(v,u);
        }
        dfs(1,-1);
        printf("Case %d: %lld\n",cnt++,ans);
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值