POJ 2054 Color a Tree

 

 

贪心。。。。

                   Color a Tree
Time Limit: 1000MS   Memory Limit: 30000K
Total Submissions: 6647   Accepted: 2249

Description

Bob is very interested in the data structure of a tree. A tree is a directed graph in which a special node is singled out, called the "root" of the tree, and there is a unique path from the root to each of the other nodes. 

Bob intends to color all the nodes of a tree with a pen. A tree has N nodes, these nodes are numbered 1, 2, ..., N. Suppose coloring a node takes 1 unit of time, and after finishing coloring one node, he is allowed to color another. Additionally, he is allowed to color a node only when its father node has been colored. Obviously, Bob is only allowed to color the root in the first try. 

Each node has a "coloring cost factor", Ci. The coloring cost of each node depends both on Ci and the time at which Bob finishes the coloring of this node. At the beginning, the time is set to 0. If the finishing time of coloring node i is Fi, then the coloring cost of node i is Ci * Fi. 

For example, a tree with five nodes is shown in Figure-1. The coloring cost factors of each node are 1, 2, 1, 2 and 4. Bob can color the tree in the order 1, 3, 5, 2, 4, with the minimum total coloring cost of 33. 

Given a tree and the coloring cost factor of each node, please help Bob to find the minimum possible total coloring cost for coloring all the nodes.

Input

The input consists of several test cases. The first line of each case contains two integers N and R (1 <= N <= 1000, 1 <= R <= N), where N is the number of nodes in the tree and R is the node number of the root node. The second line contains N integers, the i-th of which is Ci (1 <= Ci <= 500), the coloring cost factor of node i. Each of the next N-1 lines contains two space-separated node numbers V1 and V2, which are the endpoints of an edge in the tree, denoting that V1 is the father node of V2. No edge will be listed twice, and all edges will be listed. 

A test case of N = 0 and R = 0 indicates the end of input, and should not be processed. 

Output

For each test case, output a line containing the minimum total coloring cost required for Bob to color all the nodes.

Sample Input

5 1
1 2 1 2 4
1 2
1 3
2 4
3 5
0 0

Sample Output

33

Source


 

贪心原则应该是Ci大的尽量先染色,但是由于父节点染了才能染子节点的限制使得问题不好解决了,但是Ci大的一定是在其父节点染色后立即被染色,这时大牛们的思路我也没有看明白如何证明的,但仔细一想就明白了。于是我们根据这个条件就可以将Ci大的点与其父节点合并在一起组成一个集合。这样就可以将问题规模减小。

   合并后的点(即集合)的属性如何变化呢?假如设fact[i]表示集合的Ci和,iNum[i]表示i所属集合的结点个数;那么把fact[i]/iNum[i]作为贪心原则,其值大者先合并到其父节点,最终合并成一个集合。

 

复制代码
 1 #include <iostream>
 2 #include <cstring>
 3 #include <cstdio>
 4 
 5 using namespace std;
 6 
 7 struct Edge
 8 {
 9     int to,next;
10 }e[1111];
11 
12 int n,root,Size,Adj[1111],c[1111],num[1111],father[1111];
13 bool vis[1111];
14 
15 void Init()
16 {
17     Size=0;
18     memset(Adj,-1,sizeof(Adj));
19     memset(vis,false,sizeof(vis));
20 }
21 
22 void Add_Edge(int u,int v)
23 {
24     ///u-->v
25     e[Size].to=v;
26     e[Size].next=Adj[u];
27     Adj[u]=Size++;
28 }
29 
30 int Find()
31 {
32     int k=-1;
33     double maxn=-0x3f3f3f3f;
34     for(int i=1;i<=n;i++)
35     {
36         if(!vis[i]&&i!=root&&maxn<(double)c[i]/num[i])
37         {
38             maxn=(double)c[i]/num[i];
39             k=i;
40         }
41     }
42     return k;
43 }
44 
45 void Union(int a,int b)
46 {
47     /// a to b
48     num[b]+=num[a];
49     c[b]+=c[a];
50     father[a]=b;
51     for(int i=Adj[a];~i;i=e[i].next)
52     {
53         int v=e[i].to;
54         father[v]=b;
55     }
56 }
57 
58 int solve()
59 {
60     int ans=0;
61     for(int i=0;i<n-1;i++)
62     {
63         int k=Find();
64         vis[k]=true;
65         int p=father[k];
66         while(vis[p]) p=father[p];
67         ans+=c[k]*num[p];
68         Union(k,p);
69     }
70     ans+=c[root];
71     return ans;
72 }
73 
74 int main()
75 {
76     while(scanf("%d%d",&n,&root)!=EOF)
77     {
78         if(n==0&&root==0) break;
79         Init();
80         for(int i=1;i<=n;i++)
81         {
82             scanf("%d",c+i);
83             num[i]=1;
84         }
85         for(int i=0;i<n-1;i++)
86         {
87             int u,v;
88             scanf("%d%d",&u,&v);
89             Add_Edge(u,v);
90             father[v]=u;
91         }
92         printf("%d\n",solve());
93     }
94     return 0;
95 }
复制代码

 

 

posted @   码代码的猿猿  阅读(400)  评论(0编辑  收藏  举报
编辑推荐:
· 智能桌面机器人:用.NET IoT库控制舵机并多方法播放表情
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
阅读排行:
· DeepSeek火爆全网,官网宕机?本地部署一个随便玩「LLM探索」
· 开发者新选择:用DeepSeek实现Cursor级智能编程的免费方案
· 【译】.NET 升级助手现在支持升级到集中式包管理
· 独立开发经验谈:如何通过 Docker 让潜在客户快速体验你的系统
· Tinyfox 发生重大改版
点击右上角即可分享
微信分享提示