poj3140-树型dp&搜索-Contestants Division

69 篇文章 0 订阅
7 篇文章 3 订阅

https://vjudge.net/problem/17374/origin
给定一个树,问你抛去一个边,两边点权和的差最小是多少。。
好像树的重心。。开始想着枚举一遍和 重心相近的点,后来一看我去直接 枚举点 的两颗子树不就行了。。
wawawa。。
于是乎 想了好久。。原来求 反树的时候,,我用的m- siz[u]..
而其实应该计算 一下所有学生人数的(这是带权的,而不是求点的数量。。)
※ 还有就是条件没看清,开始没用longlong siz
ps:注意,求siz的函数,siz是啥呢,siz就是子树的权重大小

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long ll;
const int maxn=3e6+4000;
const int maxm=3e6+4000;
ll m;
int n;
struct Node{
      int to,next;
}node[maxn];
int len;
int head[maxn];
int vis[maxm];
ll siz[maxm];
int k[maxm];
void add(int a,int b){
     node[len].to=b;
     node[len].next=head[a];
     head[a]=len++;
}
void init(){
     len=0;
     memset(head,-1,sizeof(head));
}
ll dfs_siz(int u){
     vis[u]=true;
     siz[u]=1ll*k[u];
     for(int i=head[u];i!=-1;i=node[i].next){
         int to=node[i].to;
         if(!vis[to]){
            siz[u]+=dfs_siz(to);
         }
     }
     return siz[u];
}
int main()
{   int a,b;
   ll sum;
    int tt=1;
    while(~scanf("%lld%d",&m,&n)){
      memset(vis,false,sizeof(vis));
      memset(siz,0,sizeof(siz));
      sum=0;
      if(!m&&!n) break;
    init();
    for(int i=1;i<=m;i++){
        scanf("%d",&k[i]);
        sum+=k[i];
    }
    for(int i=0;i<n;i++){
        scanf("%d%d",&a,&b);
        add(a,b);
        add(b,a);
    }
    dfs_siz(1);
    ll  ans=2e16 ;
    ll  k;
    for(int i=1;i<=m;i++){
       if(sum-siz[i]>siz[i])
        k=1ll*(sum-siz[i]-siz[i]);
        else
        k=1ll*(siz[i]-(sum-siz[i]));
       ans=min(ans,k);
    }
    printf("Case %d: %lld\n",tt++,ans);
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值