构造随机“整数”树 - 深度优先/广度优先

时间:2009-10-14 04:06:22

标签: c recursion tree tree-traversal

这是我第一次尝试在c中编写递归函数。以下代码有效。我为长篇大论道歉,但我想尽可能清楚。

我正在尝试生成一个树,其中每个节点(inode)都有一个整数字段“n”。相应地,每个inode都有一个指向“n”其他inode的指针数组。函数inode *I = gen_tree(inode *I, int nlevels);生成一个树,每个级别具有随机数量的inode。树以深度优先的方式生成。我有几个问题。

(a)有没有更好的方法来编写函数?任何反馈/建议将不胜感激。

(b)树可以在BF时装中生成吗?

(c)I->i应该有一个遍历树的索引。如何编写一个函数来计算I->i

(d)I->c应该具有给定节点下所有inode的累积和。如何编写一个函数来计算I->c

提前致谢,

〜拉斯

//.h file:
typedef struct integerNode {
  int n;
  int c;
  int i;
  struct integerNode **nodes;
} inode;
inode *new_inode( int n );
inode *gen_itree( inode *I, int nlevels );


//Constructor:
inode *new_inode( int n ){
    inode *I;
    I = malloc( sizeof (inode ) );
    I->n = n;
    I->nodes = malloc( n * sizeof (inode* ) );
    return (I );
};

//Generating tree with random-number of nodes:
inode *gen_itree( inode *I, int nlevels ){
    int i, next_level, next_n;
    printf( " \n" );
    printf( " I : %p\n", I );
    printf( " ***** nlevels : %d\n", nlevels );
    printf( " *************\n" );
    if ( nlevels == 0 ) {
        printf( " nlevels == 0!\n");
    } else {
        printf( " I->n : %d\n", I->n );
        printf( " *************\n" );
        next_level = nlevels - 1;
        for ( i = 0; i < I->n; i++ ) {
            printf( " I: %p\n",I);
            printf( " adding node number: %d\n", i );
            next_n = 0 + rand( ) % 3;
            I->nodes[i] = new_inode( next_n );
            printf( " I->nodes[%d]->n: %p, %d\n",i, I->nodes[i],next_n);
            I->nodes[i] = gen_itree( I->nodes[i], next_level );
        }
    }
    printf( " *************\n" );
    printf( " returning I : %p\n", I );//This part is unclear to me!
    printf( " *************\n" );
    return (I);
}

//Main.c
int main( int argc, char** argv ){
    inode *I;
    I = new_inode( 2 );
    I = gen_itree(I,3);
    return ( 1 );
}

2 个答案:

答案 0 :(得分:1)

首先。您没有错误检查。你只编写了你快乐的道路。 检查你的mallocs不会返回NULL !!!

if (malloc returned NULL){
          free memory
          exit(error_code)
}

然后

 I->nodes[i] = new_inode( next_n );
 I->nodes[i] = gen_itree( I->nodes[i], next_level );

这部分内容尚不清楚。你可以这样做

 I->nodes[i] = gen_itree( new_inode( next_n ), next_level );  

同样在这里

I = new_inode( 2 );
I = gen_itree(I,3);

可能是

 I = gen_itree(new_inode( 2 ),3);

另外,别忘了释放你分配的内存。

至于(d)

unsigned int get_node_count(inode* i){
    unsigned int counter =0;

    if (!i->nodes) return 0;

     //pseudocode
     for each inode* node in i->nodes{
        counter++
        counter+= get_node_count(node);//accumulate node count in child node
     }

      return counter;

答案 1 :(得分:1)

一切看起来都不错。除非用于调试目的,否则我不会将printf放在函数中。

#define RANGE 3 // this eliminates 'magic constants'

//Generating tree with random-number of nodes:
inode *gen_itree( inode *I, int nlevels ){
        int i, next_level, next_n;

    if ( nlevels ) { // if nlevels != 0
        next_level = nlevels - 1;
        for ( i = 0; i < I->n; i++ ) {
            next_n = rand( ) % RANGE; // no need for a zero
            I->nodes[i] = new_inode( next_n );
            I->nodes[i] = gen_itree( I->nodes[i], next_level );
        }
    }

    return I;
}

看起来更好,但我甚至更进一步消除了一些不必要的局部变量,因为它们只使用一次(除了int i)。

对于(c),这应该有效:

//This computes the C's for all nodes under this, including this node  
int computeAllCs( inode *I ){
        int i;
        I->c = 0;
        for ( i = 0; i < I->n; i++ )
            I->c += computeAllCs(I->nodes[i]) + 1;
}

请注意“所有递归函数都可以迭代编写(也称为循环)”,因此您可能需要考虑迭代解决方案。