HDU 1055 贪心，并查集

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1055

题意就是怎么在规定情况下染色这个树代价最小。其实这道题，直接思考可能会不知道怎么贪心。我们不妨换一个思路。假设，我浪费一个时间啥都不染色。那么最后出来的结果必定比最小的sum大一个节点权值和。所以我们可以理解成，选中这个节点几次。为了让权值最小。权值大的我们就应该越早的选中。让sum少加这个权值次数。因为只能相邻。所以用到并查集的思想。我们将子节点和父节点合并。成一个新的节点。就可以了。

#include<bits/stdc++.h>
#define INF 1e18
#define inf 1e9
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define IOS ios_base::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std ;
typedef long long ll;
typedef unsigned long long ull;
const int _max = 1010;

struct node{
    int pre,val,t;
    double w;
};
node a[_max];
int find_max(int r,int n){
    int pos = 0;
    double cnt = -1.0;
    for(int i = 1 ; i <= n ; i++){
        if(i == r) continue;
        if(cnt < a[i].w){
            cnt = a[i].w;
            pos = i;
        }
    }
    return pos;
}
int main(){
    int n,r,u,v,ans;
    while(cin>>n>>r){
        if(!n && !r) break;
        ans = 0;
        for(int i = 1 ; i <= n ; i++){
            cin>>a[i].val;
            a[i].t = 1;
            a[i].w = a[i].val;  
            ans += a[i].val;
        }
        for(int i = 1 ; i <= n-1 ; i++){
            cin>>u>>v;
            a[v].pre = u;
        }
        for(int i = 1 ; i < n ; i++){
            int pos = find_max(r,n);
            int pre = a[pos].pre;
            ans += a[pos].val*a[pre].t;
            a[pos].w = 0;
            a[pre].t += a[pos].t;
            a[pre].val += a[pos].val;
            a[pre].w = double(a[pre].val)/double(a[pre].t);
            for(int i = 1 ; i <= n ; i++){
                if(a[i].pre == pos)
                    a[i].pre = pre;
            }
        }
        cout<<ans<<endl;    
    }
    return 0;
}