hdu3290 The magic apple tree (dfs)

题意:在一棵苹果树上,有n个节点,第一天,只有叶节点才有苹果,苹果的数目等于该节点的编号。第二天开始,其他节点开始长苹果,对于非叶节点,只有当所有直接子节点都长了苹果之后才开始长,设它的直接子节点数目为K,则该节点的苹果数等于所有直接字节点苹果数目中的第(K + 1) / 2大。问最后根节点的苹果数目。

分析:一开始感觉这题意太朦胧了,研究了半天都没明白。明白题意之后就好办很多了,类似与DP的过程,从根节点开始算,往下搜,再递推回来。

用vector实现方便很多

View Code
#include<iostream>
#include<algorithm>
#include<vector>
#define MAXN 20000+10
using namespace std;
vector<int> g[MAXN];
int n,v[MAXN],in[MAXN],out[MAXN];
int dfs(int cnt)
{
if(out[cnt]==0)//叶节点这直接返回编号
return cnt;
vector<int> ss;
for(int i=0;i<out[cnt];i++)
ss.push_back(dfs(g[cnt][i]));//保存子节点的苹果数,再算出第(k+1)/2 大
sort(ss.begin(),ss.end());
return ss[(out[cnt]+1)/2-1];
}
int main()
{
int a;
while(scanf("%d",&n)==1)
{
for(int i=1;i<=n;i++)
g[i].clear();
memset(in,0,sizeof(in));
for(int i=1;i<=n;i++)
{
scanf("%d",&out[i]);
for(int j=0;j<out[i];j++)
{
scanf("%d",&a);
g[i].push_back(a);
in[a]++;
}
}
int root;
for(int i=1;i<=n;i++)//找出根节点
if(in[i]==0)
{
root=i;
break;
}
printf("%d\n",dfs(root));
}
return 0;
}
原文地址:https://www.cnblogs.com/nanke/p/2351962.html