hdoj 3298 Contestants

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 100000000000005LL

vector<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 ;
}