题目大意:有一棵树,有一个篮子,里面有K颗石子,用石子放在树的叶子节点上,当一个中间节点的直接子节点全部都放着石子的时候,这个中间节点可以放石子,它的直接子节点上面的石子放回到篮子里,可以在后面的过程中使用,问最少需要多少颗(K)石子,可以按照要求的过程最后在根节点上放石子。
解题思路,考虑一个中间节点,假设已经求出了它的每个子节点放置石子需要的石子数量,因为数量需要大的子节点先放可以满足后面需要石子少的,所以对子节点按照need数组的值进行降序排序,这样就可以按照最节省石子的顺序去使用石子了(详情见代码),每次考虑中间节点的时候,都假想这个节点对应一个空篮子,这样递归之后,need[1]就是答案了。似乎代码更能够表达思想。
#include<iostream>
#include<cstdio>
#include<algorithm>
#define maxn 210
using namespace std;
int n,m;
int de[maxn]; //每个节点的直接子节点数目
int need[maxn]; //这个节点需要多少个石子可以使自己放上石子
int r[maxn]; //后面间接排序需要的数组
int g[maxn][maxn]; //tree存储结构
void init()
{
int node;
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d",&node);
scanf("%d",&de[node]);
for(int j=0;j<de[node];j++)
scanf("%d",&g[node][j]);
}
memset(need,0,sizeof(need));
}
void ss(int u)
{
int ed=de[u];
for(int i=0;i<ed;i++)
r[i]=i;
for(int i=0;i<ed;i++)
{
for(int j=0;j<ed-1-i;j++)
{
int va=g[u][r[j]],vb=g[u][r[j+1]];
if(need[va]<need[vb])
{
int temp=r[j];
r[j]=r[j+1];
r[j+1]=temp;
}
}
}
}
void dfs(int u)
{
int ed=de[u];
if(ed==0) //递归的界线,叶子节点只需要一个石子就可以了。
{
need[u]=1;
return ;
}
for(int i=0;i<ed;i++) //如果这个节点不是叶子节点,就递归求出子节点的need值,
dfs(g[u][i]);
ss(u); //对need进行降序,确保下面的步骤可以最优
int tbuk=0; //这个中间节点独有的一个空篮子
for(int i=0;i<ed;i++) //模拟放置石子过程,模拟完成后就求出了本节点的need值
{
int v=g[u][r[i]];
tbuk-=need[v];
if(tbuk<0)
{
need[u]+=0-tbuk;
tbuk=0;
}
tbuk+=need[v]-1;
}
}
void cal()
{
dfs(1);
printf("%d\n",need[1]);
}
void slove()
{
scanf("%d",&n);
while(n--)
{
init();
cal();
}
}
int main()
{
slove();
return 0;
}