关闭

poj 3140 Contestants Division(树dp)

466人阅读 评论(0) 收藏 举报

【题目大意】:给出一棵树,每个节点都有一个权值。删除一条边之后,问两棵子树的权值之差的最少为多少。


【解题思路】:利用dfs枚举边,记忆化搜索记录下到达当前节点并以当前节点为根的子树的权值,记为dp[i];

     则我们可以得出,两棵子树的权值之差就是abs(sum-dp[i]-dp[i])....对于每一个点去min就是答案了。


【代码】:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#include <cmath>
#include <string>
#include <cctype>
#include <map>
#include <iomanip>
                   
using namespace std;
                   
#define eps 1e-8
#define pi acos(-1.0)
#define inf 1<<30
#define linf 1LL<<60
#define pb push_back
#define lc(x) (x << 1)
#define rc(x) (x << 1 | 1)
#define lowbit(x) (x & (-x))
#define ll long long

int n,m;
ll v[100100],dp[100100],summ,ans;
vector<int> way[100100];

ll ABS(ll a){
    if (a>0) return a;
    else return -a;
}

void solve_dfs(int u,int fr){
    int vv;
    dp[u]=v[u];
    for (int i=0; i<way[u].size(); i++){
        vv=way[u][i];
        if (vv!=fr){
            solve_dfs(vv,u);
            dp[u]+=dp[vv];
            ans=min(ABS(summ-2*dp[vv]),ans);
        }
    }
    return ;
}

int main() {
    int cas=1;
    while (~scanf("%d%d",&n,&m)){
        if (n==0 && m==0) break;
        for (int i=0; i<100100; i++)
            way[i].clear();
        ans=linf;
        summ=0;
        for (int i=1; i<=n; i++){
            scanf("%lld",&v[i]);
            summ+=v[i];
        }
        int u,v;
        for (int i=0; i<m; i++){
            scanf("%d%d",&u,&v);
            way[u].pb(v);
            way[v].pb(u);
        }
        solve_dfs(1,-1);
        printf("Case %d: %lld\n",cas++,ans);
    }
    return 0;
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:95694次
    • 积分:2022
    • 等级:
    • 排名:第19720名
    • 原创:115篇
    • 转载:8篇
    • 译文:0篇
    • 评论:3条
    最新评论