借鉴别人的方法写了一下,收获挺大。
题目分析:设树中任意结点为A,当结点A为终端结点,则只需要一个石子对其进行填充;当结点a为非终端结点时,需要的石子数则由其孩子结点而定。设结点A有k个孩子(A0,A1,...,Ak-1),每个孩子需要填充时需要ai(0<=i<k)个石子,将ai存储于任意数组num[i]中,对其按降序进行排列,选出num[i]+i的最大值,即为所需石子数。
poj1694
#include
<
iostream
>
#include < algorithm >
using namespace std;
#define N1 201
int tree[N1][N1];
bool cmp( const int a , const int b )
{
return a > b;
}
int DFS( int node)
{
int num[N1], _max, i;
if ( ! tree[node][ 0 ])
return 1 ;
else
{
for (i = 1 ;i <= tree[node][ 0 ]; ++ i)
num[i - 1 ] = DFS( tree[node][i] );
sort( num , num + tree[node][ 0 ] , cmp );
for ( i = 0 , _max = - 1 ; i < tree[node][ 0 ] ; ++ i )
{
if ( _max < num[i] + i )
{ _max = num[i] + i; }
}
return _max;
}
}
int main()
{
int i,j, M,N,p;
scanf( " %d " , & M);
while (M -- )
{
scanf( " %d " , & N);
for (i = 1 ;i <= N;i ++ )
{
scanf( " %d %d " , & p, & tree[i][ 0 ]);
for (j = 1 ;j <= tree[i][ 0 ];j ++ )
scanf( " %d " , & tree[i][j]);
}
printf( " %d\n " ,DFS( 1 ));
memset(tree, 0 , sizeof (tree));
}
return 0 ;
}
#include < algorithm >
using namespace std;
#define N1 201
int tree[N1][N1];
bool cmp( const int a , const int b )
{
return a > b;
}
int DFS( int node)
{
int num[N1], _max, i;
if ( ! tree[node][ 0 ])
return 1 ;
else
{
for (i = 1 ;i <= tree[node][ 0 ]; ++ i)
num[i - 1 ] = DFS( tree[node][i] );
sort( num , num + tree[node][ 0 ] , cmp );
for ( i = 0 , _max = - 1 ; i < tree[node][ 0 ] ; ++ i )
{
if ( _max < num[i] + i )
{ _max = num[i] + i; }
}
return _max;
}
}
int main()
{
int i,j, M,N,p;
scanf( " %d " , & M);
while (M -- )
{
scanf( " %d " , & N);
for (i = 1 ;i <= N;i ++ )
{
scanf( " %d %d " , & p, & tree[i][ 0 ]);
for (j = 1 ;j <= tree[i][ 0 ];j ++ )
scanf( " %d " , & tree[i][j]);
}
printf( " %d\n " ,DFS( 1 ));
memset(tree, 0 , sizeof (tree));
}
return 0 ;
}