C格式字符串转为二叉树

　　最近在LeetCode做题,二叉树出现错误时不好排查,于是自己写了一个函数,将前序遍历格式字串转换成二叉树。

　　形如 "AB#D##C##" 的字符串，"#"表示孩子节点为空，算法如下：

　　1.当前节点进栈 push(s,t)

　　2.出栈: pcur=pop(s) ,判断当前字符
　　　　a 不等于'#'，申请新的节点pnew并赋给pcur的左或右孩子，当右孩子时将标记置真，pcur进栈，pnew进栈
　　　　b 等于'#'，如果当前为左孩子，左孩子置null，pcur进栈；
　　　　　　当前为右孩子，右孩子置空，父节点出栈。如果pcur是父节点的左孩子，终止寻找；否则保存父节点为当前节点，继续寻找。

　　代码如下：

TreeNode *creat_tree(char *str){
    if( 0 == strlen(str) ) return NULL;
    TreeNode *t = (TreeNode *)malloc(sizeof(TreeNode));
    TreeNode *pcur = NULL, *pnew = NULL;
    Stack *s = init_s();
    bool left_child = true;
    t->data = *str;
    push(s,t);
    while( '\0' != *++str ){
        pcur = pop(s);
        printf("  pcur->data:%c, val:%c\n",pcur->data,*str );        
        if( '#' != *str ){
            pnew = (TreeNode *)malloc(sizeof(TreeNode));
            pnew->data = *str;
            if( left_child ){
                pcur->lchild = pnew;
            }
            else{
                pcur->rchild = pnew;
                left_child = true;        // 下一个值要放到左孩子
            }
            push(s,pcur);            // 父节点进栈
            push(s,pnew);            // 新的结点作为下一次的父节点
        }
        else{
            if( left_child ){
                pcur->lchild = NULL;
                left_child = false;        // 无左孩子，下一个轮到右孩子
                push(s,pcur);            // 当前节点进栈，下一个轮到右孩子
            }
            else{
                pcur->rchild = NULL;    // 无右孩子
                while( ! empty_stack(s) ){
                    pnew = pop(s);        // 取出父节点
                    if( pcur != pnew->lchild ){    // 判断pcur是否为左孩子
                        pcur = pnew;            // 若不是，说明父节点已有2孩子
                    }                            // 继续查找，直到父节点只有左孩子
                    else{
                        push(s,pnew);            // 节点进栈，下一个轮到右孩子
                        break;
                    }
                }
                printf("  right end\n");
            }
        }    
    }
    return t;
}

　　测试二叉树：

void PreOrderTraverse(TreeNode *t){
    //printf("in PreOrderTraverse\n");
    if( NULL == t ) return;
    printf("%c",t->data);
    PreOrderTraverse(t->lchild);
    PreOrderTraverse(t->rchild);
}
void InOrderTraverse(TreeNode *t){
    if( NULL == t ) return;    
    InOrderTraverse(t->lchild);
    printf("%c",t->data);
    InOrderTraverse(t->rchild);
}
void PostOrderTraverse(TreeNode *t){
    if( NULL == t ) return;    
    PostOrderTraverse(t->lchild);
    PostOrderTraverse(t->rchild);
    printf("%c",t->data);
}
int main(){
    TreeNode *t = NULL;
    char str[] = "ABDH#K###E##CFI###G#J##";
    t = creat_tree(str);    
    PreOrderTraverse(t);
    printf("\n");
    InOrderTraverse(t);
    printf("\n");
    PostOrderTraverse(t);
    printf("\n");
        
    char str2[] = "AB##C##";    
    t = creat_tree(str2);    
    PreOrderTraverse(t);
    printf("\n");
    InOrderTraverse(t);
    printf("\n");
    PostOrderTraverse(t);
    printf("\n");
}

　　输出：

ABDHKECFIGJ
HKDBEAIFCGJ
KHDEBIFJGCA

ABC
BAC
BCA