最近在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( '