题意:
给出一棵一般的树,按如下的三种规则在树上放石子:
1,在游戏开始时,玩家可以拿K个石子放在水桶里。
2,在游戏的每一步,玩家可以从水桶里拿出一个石子,并放在任意一个空的叶子上。
3,当一个父节点的r个子节点都被放上了一个石子,可以将这r个石子都拿去并在父节点上放一个石子。水桶里石子可以再次使用。
显然,最终根上会被放上一个石子,问的是,求用最少的石子达到上述要求,求这个最小值。(有很多k,求最小的那个);
思路:
设所求结果用result来表示,我们要求的是以1为根的result,这个结果必定依赖于它的子节点的相关信息。这个信息会是什么呢?考虑一棵树的子孙是该树的子树。假设1的子孙分别为r1,r2,r3。那么r1,r2,r3,都是1的子树的根。假设以r1,r2,r3为根的上述结果分别为result1,result2,result3,显然它们之和一定能达到1上也能被放上石子的目的。那这个值是不是最小的呢?假设result1>=result2>=result3,我们用result1个石子让r1上放上了一个石子,此时我们剩余了result1-1个石子没有用,这些石子可以弥补result2,最终可能会得到一个更小的值(<=result1+result2+result3,通过对先前石子的重复利用)。如何安排这种弥补方式呢?
最大值优先。r1需要的石子result1最大,那么最终的结果必定result>=result1,然而最终只需在r1上放一个石子,留下了result1-1个石子。
那么r1 ,r2一共需要Result = result1+result2-min( result-1,result2 )个石子,其中min(result1-1,result2)被加了两次,所以要减掉一次。
这样r1,r2就合并成了一个节点,最少需要Result个石子,那么result=Result+result3-min(Result-2,result3)。最终得解。
显然这是一个贪心算法,符合贪心选择和最优子结构。
代码如下:
#include <iostream>
using namespace std;
const int Maxlen=200;
struct
{
int child[Maxlen];
int chNum;
}tree[Maxlen];
int cmp(const void *a,const void *b)
{
return *(int *)b-*(int *)a;
}
int min(int a,int b)
{
return a>b?b:a;
}
int least(int f)
{
int i,j,result;
int node[200];
if(!tree[f].chNum)
return 1;
for (j=i=0;i<tree[f].chNum;i++)
{
node[j++]=least(tree[f].child[i]);
}
qsort(node,j,sizeof(int),cmp);
result=node[0];
for (i=1;i<j;i++)
{
result=result+node[i]-min(result-i,node[i]);
}
return result;
}
int main()
{
int n,id,m,num_node,i;
cin>>n;
while (n--)
{
cin>>m;
while (m--)
{
cin>>id>>num_node;
tree[id].chNum=num_node;
for (i=0;i<num_node;i++)
{
cin>>tree[id].child[i];
}
}
cout<<least(1)<<endl;
}
return 0;
}