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