DAY nine 二叉树的构建和遍历

核心代码

树的初始化

BTNodePtr stringToBTree(char *paraString) {
    int i;
    char ch;
    
    QueuePtr tempQueuePtr = initQueue();
    
    BTNodePtr resultHeader;
    BTNodePtr tempParent, tempLeftChild, tempRightChild;
    i = 0;
    ch = paraString[0];
    resultHeader = constructBTNode(ch);
    enqueue(tempQueuePtr, resultHeader);
    
    while (isQueueEmpty(tempQueuePtr) == 0) {
        tempParent = dequeue(tempQueuePtr);
        
        i ++;
        ch = paraString[i];
        if (ch == '#') {
            tempParent->left = NULL;
        } else {
            tempLeftChild = constructBTNode(ch);
            enqueue(tempQueuePtr, tempLeftChild);
            tempParent->left = tempLeftChild;
        }
        
        i ++;
        ch = paraString[i];
        if (ch == '#') {
            tempParent->right = NULL;
        } else {
            tempRightChild = constructBTNode(ch);
            enqueue(tempQueuePtr, tempRightChild);
            tempParent->right = tempRightChild;
        }
    }
    
    return resultHeader;
}

先序遍历

void preorder(BTNodePtr tempPtr) {
    if (tempPtr == NULL) {
        return;
    }
    
    printf("%c", tempPtr->element);
    preorder(tempPtr->left);
    preorder(tempPtr->right);
}

中序遍历

void inorder(BTNodePtr tempPtr) {
    if (tempPtr == NULL) {
        return;
    }
    
    preorder(tempPtr->left);
    printf("%c", tempPtr->element);
    preorder(tempPtr->right);
}

后序遍历

void postorder(BTNodePtr tempPtr) {
    if (tempPtr == NULL) {
        return;
    }
    
    preorder(tempPtr->left);
    preorder(tempPtr->right);
    printf("%c", tempPtr->element);
}

总代码

#include<stdio.h>
#include<stdlib.h>
 
/**
 *树结点
 */
 
typedef struct BTNode{
    char element;
    struct BTNode *left;
    struct BTNode *right;
} BTNode, *BTNodePtr;
 
/**
 *队列
 */
 
typedef struct BTNodePtrQueue{
    BTNodePtr *nodePtrs;
    int front;
    int rear;
} BTNodePtrQueue, *QueuePtr;
 
/**
 *初始化队列 
 */
 
QueuePtr initQueue() {
    QueuePtr resultQueuePtr = (QueuePtr)malloc(sizeof(BTNodePtrQueue));
    resultQueuePtr->nodePtrs = (BTNodePtr*)malloc(sizeof(BTNode) * 5);
    resultQueuePtr->front = 0;
    resultQueuePtr->rear = 1;
    
    return resultQueuePtr;
}
 
/**
 *判断是否为空
 */
 
int isQueueEmpty(QueuePtr paraQueuePtr) {
    if ((paraQueuePtr->front + 1) % 5 == paraQueuePtr->rear) {
        return 1;
    }
    
    return 0;
}
 
/**
 *添加元素到队列
 */
 
void enqueue(QueuePtr paraQueuePtr, BTNodePtr paraBTNodePtr) {
    printf("front = %d, rear = %d.\n", paraQueuePtr->front, paraQueuePtr->rear);
    if ((paraQueuePtr->rear + 1) % 5 == paraQueuePtr->front % 5) {
        printf("Error, trying to enqueue %c. queue full\n", paraBTNodePtr->element);
        return;
    }
    paraQueuePtr->nodePtrs[paraQueuePtr->rear] = paraBTNodePtr;
    paraQueuePtr->rear = (paraQueuePtr->rear + 1) % 5;
    printf("enqueue %c ends.\n", paraBTNodePtr->element);
}
 
/**
 *删除
 */
 
BTNodePtr dequeue(QueuePtr paraQueuePtr) {
    if (isQueueEmpty(paraQueuePtr) == 1) {
        printf("Error, empty queue\n");
        return NULL;
    }
    
    paraQueuePtr->front = (paraQueuePtr->front + 1) % 5;
    
    printf("dequeue %c ends.\n", paraQueuePtr->nodePtrs[paraQueuePtr->front]->element);
    return paraQueuePtr->nodePtrs[paraQueuePtr->front];
}
 
/**
 *初始化结点
 */
 
BTNodePtr constructBTNode(char paraChar) {
    BTNodePtr resultPtr = (BTNodePtr)malloc(sizeof(BTNode));
    resultPtr->element = paraChar;
    resultPtr->left = NULL;
    resultPtr->right = NULL;
    return resultPtr;
}
 
/**
 *初始化树 
 */
 
BTNodePtr stringToBTree(char *paraString) {
    int i;
    char ch;
    
    QueuePtr tempQueuePtr = initQueue();
    
    BTNodePtr resultHeader;
    BTNodePtr tempParent, tempLeftChild, tempRightChild;
    i = 0;
    ch = paraString[0];
    resultHeader = constructBTNode(ch);
    enqueue(tempQueuePtr, resultHeader);
    
    while (isQueueEmpty(tempQueuePtr) == 0) {
        tempParent = dequeue(tempQueuePtr);
        
        i ++;
        ch = paraString[i];
        if (ch == '#') {
            tempParent->left = NULL;
        } else {
            tempLeftChild = constructBTNode(ch);
            enqueue(tempQueuePtr, tempLeftChild);
            tempParent->left = tempLeftChild;
        }
        
        i ++;
        ch = paraString[i];
        if (ch == '#') {
            tempParent->right = NULL;
        } else {
            tempRightChild = constructBTNode(ch);
            enqueue(tempQueuePtr, tempRightChild);
            tempParent->right = tempRightChild;
        }
    }
    
    return resultHeader;
}
 
/**
 *层序遍历
 */
  
void levelwise(BTNodePtr paraTreePtr) {
    char tempString[100];
    int i = 0;
    QueuePtr tempQueuePtr = initQueue();
    BTNodePtr tempNodePtr;
    enqueue(tempQueuePtr, paraTreePtr);
    while (isQueueEmpty(tempQueuePtr) == 0) {
        tempNodePtr = dequeue(tempQueuePtr);
        
        tempString[i] = tempNodePtr->element;
        i ++;
        
        if (tempNodePtr->left != NULL) {
            enqueue(tempQueuePtr, tempNodePtr->left);
        }
        
        if (tempNodePtr->right != NULL) {
            enqueue(tempQueuePtr, tempNodePtr->right);
        }
    }
    tempString[i] = '\0';
    
    printf("Levelwise:%s\n", tempString);
}
 
/**
 *先序
 */
 
void preorder(BTNodePtr tempPtr) {
    if (tempPtr == NULL) {
        return;
    }
    
    printf("%c", tempPtr->element);
    preorder(tempPtr->left);
    preorder(tempPtr->right);
}
 
/**
 *中序
 */
 
void inorder(BTNodePtr tempPtr) {
    if (tempPtr == NULL) {
        return;
    }
    
    preorder(tempPtr->left);
    printf("%c", tempPtr->element);
    preorder(tempPtr->right);
}
 
/**
 *后序
 */
 
void postorder(BTNodePtr tempPtr) {
    if (tempPtr == NULL) {
        return;
    }
    
    preorder(tempPtr->left);
    preorder(tempPtr->right);
    printf("%c", tempPtr->element);
}
 
int main() {
    BTNodePtr tempHeader;
    tempHeader = constructBTNode('c');
    printf("There is only one node. \nPreorder visit:");
    preorder(tempHeader);
    printf("\n");
    
    char *tempString = "acde#bf######";
    
    tempHeader = stringToBTree(tempString);
    printf("Preorder:");
    preorder(tempHeader);
    printf("\n");
    printf("Inorder:");
    inorder(tempHeader);
    printf("\n");
    printf("Postorder:");
    postorder(tempHeader);
    printf("\n");
    printf("Levelwise:");
    levelwise(tempHeader);
    printf("\n");
    
    return 0;
}

运算结果