1

我试图查看树 A 是否是树 B 的前序遍历，为此，我创建了两棵树，它们在遍历期间应该保存值。但是，经过数小时的调试后，我发现树的值始终为 0。我对为什么树的值为 0 感到困惑。我已经完成了大量的打印语句（其中一些我留在了发布的代码中下面），我似乎无法确定为什么会发生这种情况。有人可以将我推向正确的方向吗？我想也许该函数在退出时正在删除变量，为了解决问题，我让 preorder 函数返回树（如下所示），但是，树的输出始终为 0。

代码：

typedef struct node
{
    // Each node holds a single integer.
    int data;

    // Pointers to the node's left and right children.
    struct node *left, *right;
} node;
int tree_diff(node *a, node *b)
{
    if (a == NULL && b == NULL)
        return 0;

    else if (a == NULL || b == NULL)
        return 1;


    else if (a->data != b->data)
        return 1;
    printf("A %d , B %d", a->data, b->data);

        return tree_diff(a->left, b->left) || tree_diff(a->right, b->right);
}

node *preorder_recursive(node *root, node *A)
{
    if (root == NULL)
        return A;

    printf("root %d ", root->data);
    A = root;
    printf("= data %d\n", A->data);
    preorder_recursive(root->left, A->left);
    preorder_recursive(root->right, A->right);
}

void postorder_recursive(node *root, node *B)
{
    if (root == NULL)
        return;
    B = root;
    postorder_recursive(root->left, B->left);
    postorder_recursive(root->right, B->right);
    printf("root %d ", root->data);
    printf("= data %d\n", B->data);
}

int kindredSpirits(node *a, node *b)
{

  // Get the preorder of a
  node *A = malloc(sizeof(node));
  A = preorder_recursive(a, A);

  // Get the postorder of b
  printf("\n\n");
  node *B = malloc(sizeof(node));
  postorder_recursive(b, B);

if(tree_diff(A,B) == 1)
  return 0;
else
  return 1;
}

测试用例： 

#include <stdio.h>
#include <stdlib.h>
#include "KindredSpirits.h"
  node *create_node(int data)
{
    node *n = malloc(sizeof(node));

    n->data = data;
    n->left = n->right = NULL;

    return n;
}

node *forest_fire(node *root)
{
    if (root == NULL)
        return NULL;

    forest_fire(root->left);
    forest_fire(root->right);
    free(root);
}

int main(void)
{
    node *root;

    root = create_node(23);
    root->left = create_node(12);
    root->left->left = create_node(5);
    root->left->right = create_node(18);
    root->right = create_node(71);
    root->right->right = create_node(56);

    printf("%s\n", !kindredSpirits(root, root) ? "Success!" : "fail whale :(");

    forest_fire(root);

    return 0;
}
4

2 回答 2

1

我正在尝试编写一个函数来确定树 a 的前序是否等于树 b 的后序而不改变 a 或 b。

我使用数组来保存遍历，然后比较两个数组的值。

#define MAX_TREE_SIZE 100

void preorder_recursive(node *root, int* arr, int *len) {
  if (root != NULL){
    arr[(*len)++] = root->data;
    preorder_recursive(root->left, arr, len);
    preorder_recursive(root->right, arr, len);
  }
}

void postorder_recursive(node *root, int *arr, int *len) {
  if (root != NULL){
   postorder_recursive(root->left, arr, len);
   postorder_recursive(root->right, arr, len);
   arr[(*len)++] = root->data;
  }
}

int kindredSpirits(node *a, node *b){

 // Get the preorder of a
 int *arr1 = malloc(MAX_TREE_SIZE * sizeof(int));
 int len1 = 0;
 preorder_recursive(a, arr1, &len1);

 // Get the postorder of b
 int *arr2 = malloc(MAX_TREE_SIZE * sizeof(int));
 int len2 = 0;
 postorder_recursive(b, arr2, &len2);

 int ret = 0; // 2 traversals are equal
 if (len1 != len2) {
    ret = 1;
 } else {
   for (int i = 0; i < len1; i++){
     if (arr1[i] != arr2[i]){
       ret = 1;
       break;
     }
   }
 }

 free(arr1);
 free(arr2);
 return ret;
}
于 2017-04-03T02:45:26.397 回答
1

这是一个可以帮助您入门的代码片段：

typedef struct node {
    // Each node holds a single integer.
    int data;

    // Pointers to the node's left and right children.
    struct node *left,
    struct node *right;
} node;

typedef struct list {
    int lst_max;                        // maximum number of allocated cells
    int lst_cur;                        // current number of filled cells
    int *lst_base;                      // traversal list
} list;

list list_a = { 0, 0, NULL };
list list_b = { 0, 0, NULL };

void
list_append(list *lst,int data)
{
    int newidx;

    newidx = lst->lst_cur;

    if (newidx >= lst->lst_max) {
        lst->lst_max += 100;
        lst->lst_base = realloc(lst->lst_base,sizeof(int) * lst->lst_max);
        if (lst->lst_base == NULL) {
            printf("list_append: malloc error\n");
            exit(1);
        }
    }

    lst->lst_base[newidx] = data;

    lst->lst_cur = newidx + 1;
}

void
preorder_recursive(node *root,list *lst)
{

    if (root == NULL)
        return;

    list_append(lst,root->data);
    preorder_recursive(root->left,lst);
    preorder_recursive(root->right,lst);
}

void
postorder_recursive(node *root,list *lst)
{

    if (root == NULL)
        return;

    postorder_recursive(root->left,lst);
    postorder_recursive(root->right,lst);
    list_append(lst,root->data);
}

int
main(void)
{

    preorder_recursive(a,&list_a);
    postorder_recursive(b,&list_b);

    // compare list_a and list_b ...

    return 0;
}
于 2017-04-03T02:51:00.197 回答