POJ 2054 贪心 !!! Color a Tree

http://poj.org/problem?id=2054

题意:给出n个节点和r根节点   给这些节点涂色,除根节点外每个节点都有一个父节点( and there is a unique path from the root to each of the other nodes. ),每涂一个节点花费一个单位时间,涂子节点;每个节点都有权值ci,  涂完一个节点后花费的时间ti,那么该节点的花费为ci*ti,题目要求按怎样的顺序涂色花费的最少,求出最少的花费

 开始觉得贪心肯定不行,因为每次找出最大的,局部优化不一定达到最终优化,至于搜索,广搜肯定不行,深搜也现实。最后搜到了解题报告,是巧妙的贪心,把子节点和父节点合成一个节点(mygod,打死也想不出来这种方法),知道思路写代码还是问题,我想先把路径找出来再计算,结果做不到。天,解题报告中,边贪心边计算实在是巧妙。总之,说好听了,这个题都是学习别人的

代码:

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
struct
{
    int c;
    double w;
    int node;//子节点数
    int f;//父节点

}Node[1005];
int main()
{
    int N,i,root,SUM;
    while(scanf("%d%d",&N,&root)&&N&&root)
    {
       SUM=0;
       for(i=1;i<=N;i++)
       {
          scanf("%d",&Node[i].c);
          Node[i].w=Node[i].c;
          Node[i].node=1;
          SUM+=Node[i].c;
       }
       int son,father;
       for(i=1;i<N;i++)
       {
           scanf("%d%d",&father,&son);
           Node[son].f=father;
       }

       int nn=N-1;
       int ff;
       while(nn--)
       {
           int mark;
           double max=-1.0;
           for(i=1;i<=N;i++)
           {
               if(max<Node[i].w&&i!=root)//why??  开始把i!=root放到for 条件里面了,是抄晕了吧  哈哈  怎么这样办呢
               {
                   max=Node[i].w;
                   mark=i;
               }
           }
           Node[mark].w=0;
           ff=Node[mark].f;
           SUM+=Node[mark].c*Node[ff].node;
           Node[ff].node+=Node[mark].node;//Node[ff].node+=1;
           Node[ff].c+=Node[mark].c;
           Node[ff].w=(double)Node[ff].c/Node[ff].node;

           for(i=1;i<=N;i++)
           {
               if(Node[i].f==mark)
               Node[i].f=ff;
           }
       }
       printf("%d\n",SUM);

    }
    return 0;
}

  

posted @ 2012-01-19 00:20  快乐.  阅读(245)  评论(0编辑  收藏  举报