poj-2054 Color a Tree ****

  1 /*
  2  * POJ-2054 color a tree
  3  *
  4  *    贪心 难题！
  5  *
  6  *    思路1：
  7  *    http://hi.baidu.com/cheezer94/blog/item/d98eca065202a2f237d122da.html
  8  *    思路2：
  9  *    http://www.cnblogs.com/X-Kly/archive/2011/11/02/POJ2054.html
 10  *    代码：
 11  *    http://xinbaolianmeng.com/showshouye.aspx?type=7&ID=110
 12  *
 13  *    简单讲：
 14  *        每次找到 权值最大的“集合”， 与其“父集合”合并， 权值为  两个集合中 顶点权值的平均。
 15  *        直到只剩一个集合
 16  *
 17  *
 18  *    下面有范例代码， 一样， 可看其注释
 19  *
 20  *  
 21  */
 22 
 23 
 24 #include <cstdio>
 25 using namespace std;
 26 
 27 const int maxn = 1000 + 5;
 28 
 29 int n, r;
 30 int c[maxn], fa[maxn];
 31 int pre[maxn], next[maxn], sum[maxn], num[maxn];
 32 bool vis[maxn];
 33 
 34 int find_max(){
 35     double max = -1;
 36     int maxNum = -1;
 37     for(int i=1; i<=n; i++){
 38         if(!vis[i] && (sum[i] * 1.0 / num[i]) > max){
 39             max = (sum[i] * 1.0 / num[i]);
 40             maxNum = i;
 41         }
 42     }
 43     return maxNum;
 44 }
 45 
 46 
 47 void unionFa(int x){
 48     int faSet;
 49     for(faSet=fa[x]; pre[faSet]!=-1; faSet = pre[faSet]);
 50     sum[faSet] += sum[x];
 51     num[faSet] += num[x];
 52     for(faSet=fa[x]; next[faSet]!=-1; faSet = next[faSet]);
 53     next[faSet] = x;
 54     pre[x] = faSet;
 55 
 56     vis[x] = 1;
 57 }
 58 
 59 int main(){
 60     while(scanf("%d%d", &n, &r)){
 61         if(n == 0) return 0;
 62 
 63         for(int i=1; i<=n; i++){                //注意从1开始！！（因为输入的根、边的标号都是从1开始的）
 64             scanf("%d", &c[i]);
 65             sum[i] = c[i];
 66             num[i] = 1;
 67             pre[i] = next[i] = -1;
 68             vis[i] = false;
 69         }
 70 
 71         int u, v;
 72         for(int i=1; i<n; i++){
 73             scanf("%d%d", &u, &v);
 74             fa[v] = u;
 75         }
 76 
 77         int d;
 78         vis[r] = 1;
 79         while(1){
 80             d = find_max();
 81             if(d == -1) break;
 82             unionFa(d);
 83         }
 84 
 85         int ans = 0, t = 1;
 86         for(int i=r; i!=-1; i=next[i]){
 87             ans += t * c[i];
 88             t++;
 89         }
 90         printf("%d\n", ans);
 91     }
 92 
 93 
 94     return 0;
 95 }
 96 
 97 /* ===================== 范例程序 有注释 ================================
 98 
 99     #include<iostream>
100 　　#include <cstdio>
101     using namespace std;
102     const int max_n=1010;
103     int pre[max_n];//pre[i]用来表示当前集合（包含i）的上个元素
104     int next[max_n];//next[i]用来表示当前集合（包含i）的下个元素
105     int c[max_n];
106     int num[max_n];//num[i]表示当前集合（包含i）的元素个数
107     int visit[max_n];
108     int sum[max_n];//当前集合元素和
109     int father[max_n];//father[i]表示i的父元素
110     int n,r;
111     int find_max()//找到当前权值最大的集合，合并后的点当做一个集合，也就是一个点
112     {
113         double max=0;
114         int bh=-1;
115         for(int i=1;i<=n;i++)
116         {
117             if(max<(sum[i]*1.0)/num[i]&&visit[i]==0)
118             {
119                  max=(sum[i]*1.0)/num[i];
120                  bh=i;
121             }
122         }
123         return bh;
124     }
125     void uni(int x)//联合
126     {
127         int i;
128         for(i = father[x]; pre[i] != -1; i = pre[i]);//找到父元素所在的集合
129         sum[i]+=sum[x];
130         num[i]+=num[x];
131         for(i = father[x]; next[i] != -1; i = next[i]);//找到父元素所在集合的底元素
132         next[i]=x;
133         pre[x]=i;
134         visit[x]=1;
135     }
136     int main()
137     {
138 
139          while(scanf("%d %d",&n,&r),n&&r)
140          {
141          for(int i=1;i<=n;i++)
142          {
143             scanf("%d",&c[i]);
144             sum[i]=c[i];
145             visit[i] =0;
146             pre[i] = next[i] = -1;
147             num[i] = 1;
148          }
149          for(int i = 1; i < n; i++)
150             {
151                 int a, b;
152                 scanf("%d %d", &a, &b);
153                 father[b] = a;
154             }
155 
156          int d;
157          visit[r]=1;
158          while(1)
159          {
160              d=find_max();
161              if(d==-1)break;
162              uni(d);
163          }
164          int ans=0,tjq=1;
165          for(int i=r;i!=-1;i=next[i])
166          {
167             ans+=tjq*c[i];
168             tjq++;
169          }
170          printf("%d\n",ans);
171          }
172 
173          return 0;
174     }
175 
176 */