【dp每一天】 POJ - 3140 Contestants Division (树形dp)

这个故事告诉我们 long long的abs需要自己写:)

说起来紫书上管这个点叫树的重心?看来还get了重心的求法...

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;
#define ll long long
const int inf = 0x3f3f3f3f;
const int maxn = 1e5+20, maxe = 1e6+10;
int n,m;
struct node
{
    int to,next;
    node(){}
    node(int a,int b){to = a; next = b;}
}edge[maxe << 1];
int h[maxn],vis[maxn];
ll k[maxn];
ll sum[maxn];
int edgenum;
ll ans = 1e14,num;
void init()
{
    for(int i = 1; i <= n ; i++)
        h[i] = -1, k[i] = vis[i] = sum[i] = 0;
    edgenum = 0;    ans = 1e15;  num = 0;
}

void add(int f,int t)
{
    edge[edgenum] = node(t,h[f]);
    h[f] = edgenum++;
}
void solve(int u)
{
    sum[u] = k[u];
    for(int i = h[u] ;~i; i = edge[i].next)
    {
        int v = edge[i].to;
        if(vis[v]) continue;
        //cout <<"u v "<<u << " " << v << endl;
        vis[v] = 1;
        solve(v);
        sum[u] += sum[v];
    }
    ll th = num-sum[u]-sum[u] >= 0? num-sum[u]-sum[u]:(-(num-sum[u]-sum[u]));
    ans = min(ans,th);
}
int main()
{
    int kase = 1;int a,b;
    while(~scanf("%d%d",&n,&m) && n+m)
    {
        init();
        for(int i = 1; i <= n; i++)  {scanf("%lld",&k[i]);num += k[i];}
        //cout << num << endl;
        for(int i = 0; i < m ;i++)
        {
            scanf("%d%d",&a,&b);
            add(a,b); add(b,a);
        }

        vis[1] = 1;ans = num;
        solve(1);
        //for(int i = 1; i <= n ; i++)
        //    cout <<"i " << i << " " << sum[i] << endl;

        printf("Case %d: %lld\n",kase++,ans);
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值