Question

我使用堆栈进行非递归后序。但是在Main函数中，我不想手工输入所有TreeNodes，而是想使用scanf获取TreeNodes。例如，printf（＆＃34;你想要多少个TreeNodes？＆＃34;）;并获取TreeNodes的数量，然后获取printf（＆＃34;输入TreeNodes＆＃34;）;所以我像这样输入1 2 3 4 5。我该如何编码？

#include <stdio.h>
#include <stdbool.h>
#define STACK_SIZE 10
typedef struct TreeNode {
    int data;
    struct TreeNode* left;
    struct TreeNode* right;
}TreeNode;

typedef struct Stack {
    TreeNode *buf[STACK_SIZE];
    int top;
}Stack;


void postOrder(TreeNode *root, Stack *stack)
{
    Stack* s = stack;
    if (root == NULL) return;
    TreeNode* current = root;
    TreeNode *tmp;
    bool done = 0;
    InitStack(s);


    while(!done)
    {
        while (current != NULL)
        {
            if (current->right != NULL)
                Push(s, current->right);
            Push(s, current);
            current = current->left;
        }
        if (IsEmpty(s))
            break;
        current = Pop(s);
        if (IsEmpty(s))
        {
            printf("%d", current->data);
            break;
        }
        tmp = Pop(s);
        if (tmp == current->right)
        {
            Push(s, current);
            current = current->right;
        }
        else
        {
            printf("%d", current->data);
            Push(s, tmp);
            current = NULL;
        }
    }
}
int main()
{   
    Stack s;
    TreeNode one, two, three, four, five;

    one.data = 1;
    two.data = 2;
    three.data = 3;
    four.data = 4;
    five.data = 5;

    one.left = &two;        one.right = &three;
    two.left = &four;       two.right = &five;
    three.left = NULL;      three.right = NULL;
    four.left = NULL;       four.right = NULL;
    five.left = NULL;       five.right = NULL;

    postOrder(&one, &s);

    printf("\n");
    getchar();

    return 0;
}

Answer 1

使用动态内存分配实现单独的方法来构造树。使用malloc函数分配内存。将新节点插入到适当位置的树中。

void insertNode(TreeNode *root,int data){
        //To Do create node dynamically using malloc and attach to root at appropriate position
    }

读取输入的方法如下所示：

int main()
{
    Stack s;
    TreeNode *rootNode = NULL;
    int nodeCount;
    int data;
    printf("Enter number of nodes:\n");
    scanf("%d",&nodeCount);
    printf("enter TreeNodes Data:\n");
    for (int i=0; i < nodeCount; i++) {
        scanf("%d",&data);
        insertNode(rootNode,data);
    }
    postOrder(rootNode, &s);

    return 0;
}

获取二叉树节点

1 个答案: