hdoj 3298 Contestants

来源：互联网发布：开电话会议用什么软件编辑：程序博客网时间：2024/05/29 13:47

Contestants

树形DP题目，这个题目与hduoj4081以及金华邀请赛的power station (POJ 4045)比较相似。不过这个题目相对于后面两个要简单的多。我们可以先进行一遍DFS处理出以1为根节点的树的所有节点的子节点数目，然后就是求最小差值。枚举每条边，这里可以直接枚举点来进行。注意数据范围就可以了。

#include <iostream>#include <string.h>#include <stdio.h>#include <vector>using namespace std ;#define MAXN 100005#define INF 100000000000005LLvector<int> G[MAXN] ;int n ;int m ;int num[MAXN] ;long long sum[MAXN] ;long long nmin ;void dfs1(int far , int v){    sum[v] +=(long long) num[v] ;    for(int i = 0 ; i <G[v].size() ; i ++){        int u = G[v][i] ;        if(u!=far){            dfs1(v , u) ;            sum[v] += sum[u] ;        }    }}void dfs2(int far , int v){    for(int i = 0 ; i < G[v].size() ; i ++){        int u = G[v][i] ;        if(far != u){            long long tmp ;            tmp = (long long)(sum[v] - sum[u]) > sum[u] ? (sum[v] - 2*sum[u]) : (2*sum[u] - sum[v]) ;            nmin = min(nmin , tmp) ;            sum[u] = sum[v] ;            dfs2(v , u) ;        }    }}void init(){    for(int i = 1 ; i <= n ; i ++){        G[i].clear() ;    }    memset(sum , 0 , sizeof(sum)) ;}int main(){    int T ;    T = 0 ;    while(scanf("%d%d" , &n , &m) , m + n){        init() ;        for(int i = 1 ; i <= n ; i ++){            scanf("%d" , &num[i]) ;        }        int a ;        int b ;        for(int i = 1 ; i <= m ; i ++){            scanf("%d%d" , &a , &b) ;            G[a].push_back(b) ;            G[b].push_back(a) ;        }        nmin = INF ;        dfs1(0 , 1) ;        dfs2(0 , 1) ;        printf("Case %d: %I64d\n" , ++T , nmin) ;    }    return 0 ;}