版权所有,转载请注明出处,谢谢!
http://blog.csdn.net/walkinginthewind/article/details/7518888
树是一种比较重要的数据结构,尤其是二叉树。二叉树是一种特殊的树,在二叉树中每个节点最多有两个子节点,一般称为左子节点和右子节点(或左孩子和右孩子),并且二叉树的子树有左右之分,其次序不能任意颠倒。二叉树是递归定义的,因此,与二叉树有关的题目基本都可以用递归思想解决,当然有些题目非递归解法也应该掌握,如非递归遍历节点等等。本文努力对二叉树相关题目做一个较全的整理总结,希望对找工作的同学有所帮助。
二叉树节点定义如下:
struct BinaryTreeNode
{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
};
相关链接:
轻松搞定面试中的链表题目
题目列表:
1. 求二叉树中的节点个数
2. 求二叉树的深度
3. 前序遍历,中序遍历,后序遍历
4.分层遍历二叉树(按层次从上往下,从左往右)
5. 将二叉查找树变为有序的双向链表
6. 求二叉树第K层的节点个数
7. 求二叉树中叶子节点的个数
8. 判断两棵二叉树是否结构相同
9. 判断二叉树是不是平衡二叉树
10. 求二叉树的镜像
11. 求二叉树中两个节点的最低公共祖先节点
12. 求二叉树中节点的最大距离
13. 由前序遍历序列和中序遍历序列重建二叉树
14.判断二叉树是不是完全二叉树
详细解答
1. 求二叉树中的节点个数
递归解法:
(1)如果二叉树为空,节点个数为0
(2)如果二叉树不为空,二叉树节点个数 = 左子树节点个数 + 右子树节点个数 + 1
参考代码如下:
- int GetNodeNum(BinaryTreeNode * pRoot)
- {
- if(pRoot == NULL)
- return 0;
- return GetNodeNum(pRoot->m_pLeft) + GetNodeNum(pRoot->m_pRight) + 1;
- }
2. 求二叉树的深度
递归解法:
(1)如果二叉树为空,二叉树的深度为0
(2)如果二叉树不为空,二叉树的深度 = max(左子树深度, 右子树深度) + 1
参考代码如下:
- int GetDepth(BinaryTreeNode * pRoot)
- {
- if(pRoot == NULL)
- return 0;
- int depthLeft = GetDepth(pRoot->m_pLeft);
- int depthRight = GetDepth(pRoot->m_pRight);
- return depthLeft > depthRight ? (depthLeft + 1) : (depthRight + 1);
- }
3. 前序遍历,中序遍历,后序遍历
前序遍历递归解法:
(1)如果二叉树为空,空操作
(2)如果二叉树不为空,访问根节点,前序遍历左子树,前序遍历右子树
参考代码如下:
- void PreOrderTraverse(BinaryTreeNode * pRoot)
- {
- if(pRoot == NULL)
- return;
- Visit(pRoot);
- PreOrderTraverse(pRoot->m_pLeft);
- PreOrderTraverse(pRoot->m_pRight);
- }
中序遍历递归解法
(1)如果二叉树为空,空操作。
(2)如果二叉树不为空,中序遍历左子树,访问根节点,中序遍历右子树
参考代码如下:
- void InOrderTraverse(BinaryTreeNode * pRoot)
- {
- if(pRoot == NULL)
- return;
- InOrderTraverse(pRoot->m_pLeft);
- Visit(pRoot);
- InOrderTraverse(pRoot->m_pRight);
- }
后序遍历递归解法
(1)如果二叉树为空,空操作
(2)如果二叉树不为空,后序遍历左子树,后序遍历右子树,访问根节点
参考代码如下:
- void PostOrderTraverse(BinaryTreeNode * pRoot)
- {
- if(pRoot == NULL)
- return;
- PostOrderTraverse(pRoot->m_pLeft);
- PostOrderTraverse(pRoot->m_pRight);
- Visit(pRoot);
- }
4.分层遍历二叉树(按层次从上往下,从左往右)
相当于广度优先搜索,使用队列实现。队列初始化,将根节点压入队列。当队列不为空,进行如下操作:弹出一个节点,访问,若左子节点或右子节点不为空,将其压入队列。
- void LevelTraverse(BinaryTreeNode * pRoot)
- {
- if(pRoot == NULL)
- return;
- queue<BinaryTreeNode *> q;
- q.push(pRoot);//添加元素到队列尾部
- while(!q.empty())
- {
- BinaryTreeNode * pNode = q.front();//指向队列的首部
- q.pop();//删除首部元素(即指针)
- Visit(pNode);
- if(pNode->m_pLeft != NULL)
- q.push(pNode->m_pLeft);//添加元素到队列尾部
- if(pNode->m_pRight != NULL)
- q.push(pNode->m_pRight);//添加元素到队列尾部
- }
- return;
- }
5. 将二叉查找树变为有序的双向链表
要求不能创建新节点,只调整指针。
递归解法:
(1)如果二叉树查找树为空,不需要转换,对应双向链表的第一个节点是NULL,最后一个节点是NULL
(2)如果二叉查找树不为空:
如果左子树为空,对应双向有序链表的第一个节点是根节点,左边不需要其他操作;
如果左子树不为空,转换左子树,二叉查找树对应双向有序链表的第一个节点就是左子树转换后双向有序链表的第一个节点,同时将根节点和左子树转换后的双向有序链
表的最后一个节点连接;
如果右子树为空,对应双向有序链表的最后一个节点是根节点,右边不需要其他操作;
如果右子树不为空,对应双向有序链表的最后一个节点就是右子树转换后双向有序链表的最后一个节点,同时将根节点和右子树转换后的双向有序链表的第一个节点连
接。
参考代码如下:
-
-
-
-
-
-
- void Convert(BinaryTreeNode * pRoot,
- BinaryTreeNode * & pFirstNode, BinaryTreeNode * & pLastNode)
- {
- BinaryTreeNode *pFirstLeft, *pLastLeft, * pFirstRight, *pLastRight;
- if(pRoot == NULL)
- {
- pFirstNode = NULL;
- pLastNode = NULL;
- return;
- }
-
- if(pRoot->m_pLeft == NULL)
- {
-
- pFirstNode = pRoot;
- }
- else
- {
- Convert(pRoot->m_pLeft, pFirstLeft, pLastLeft);
-
- pFirstNode = pFirstLeft;
-
- pRoot->m_pLeft = pLastLeft;
- pLastLeft->m_pRight = pRoot;
- }
-
- if(pRoot->m_pRight == NULL)
- {
-
- pLastNode = pRoot;
- }
- else
- {
- Convert(pRoot->m_pRight, pFirstRight, pLastRight);
-
- pLastNode = pLastRight;
-
- pRoot->m_pRight = pFirstRight;
- pFirstRight->m_pLeft = pRoot;
- }
-
- return;
- }
6. 求二叉树第K层的节点个数
递归解法:
(1)如果二叉树为空或者k<1返回0
(2)如果二叉树不为空并且k==1,返回1
(3)如果二叉树不为空且k>1,返回左子树中k-1层的节点个数与右子树k-1层节点个数之和
参考代码如下:
- int GetNodeNumKthLevel(BinaryTreeNode * pRoot, int k)
- {
- if(pRoot == NULL || k < 1)
- return 0;
- if(k == 1)
- return 1;
- int numLeft = GetNodeNumKthLevel(pRoot->m_pLeft, k-1);
- int numRight = GetNodeNumKthLevel(pRoot->m_pRight, k-1);
- return (numLeft + numRight);
- }
7. 求二叉树中叶子节点的个数
完全二叉树有2*n-1 个节点,则它的叶子节点数为?
完全二叉树的节点数是奇数,说明此完全二叉树也是满二叉树,也就是说每个内部节点正好都有2个叶结点.
设内部节点数为a,叶节点数为b,结点总数为m,明显有a+b=m (1)
非空满二叉树中所有节点的出度正好等于入度,每个内部节点出度为2,叶节点出度为0,所有节点的出度和为2a;根节点入度为0,其他节点的入度为1,所有节点的入度和为a+b-1;因此有2a=a+b-1 (2)
由(1),(2)得 b=(m+1)/2,a=(m-1)/2,b=a+1
也就是说,非空满二叉树的叶节点数正好比内部节点数多1
此完全二叉树的结点总数为2n-1,因此其叶结点数为n.
递归解法:
(1)如果二叉树为空,返回0
(2)如果二叉树不为空且左右子树为空,返回1
(3)如果二叉树不为空,且左右子树不同时为空,返回左子树中叶子节点个数加上右子树中叶子节点个数
参考代码如下:
- int GetLeafNodeNum(BinaryTreeNode * pRoot)
- {
- if(pRoot == NULL)
- return 0;
- if(pRoot->m_pLeft == NULL && pRoot->m_pRight == NULL)
- return 1;
- int numLeft = GetLeafNodeNum(pRoot->m_pLeft);
- int numRight = GetLeafNodeNum(pRoot->m_pRight);
- return (numLeft + numRight);
- }
8. 判断两棵二叉树是否结构相同
不考虑数据内容。结构相同意味着对应的左子树和对应的右子树都结构相同。
递归解法:
(1)如果两棵二叉树都为空,返回真
(2)如果两棵二叉树一棵为空,另一棵不为空,返回假
(3)如果两棵二叉树都不为空,如果对应的左子树和右子树都同构返回真,其他返回假
参考代码如下:
- bool StructureCmp(BinaryTreeNode * pRoot1, BinaryTreeNode * pRoot2)
- {
- if(pRoot1 == NULL && pRoot2 == NULL)
- return true;
- else if(pRoot1 == NULL || pRoot2 == NULL)
- return false;
- bool resultLeft = StructureCmp(pRoot1->m_pLeft, pRoot2->m_pLeft);
- bool resultRight = StructureCmp(pRoot1->m_pRight, pRoot2->m_pRight);
- return (resultLeft && resultRight);
- }
9. 判断二叉树是不是平衡二叉树
递归解法:
(1)如果二叉树为空,返回真
(2)如果二叉树不为空,如果左子树和右子树都是AVL树并且左子树和右子树高度相差不大于1,返回真,其他返回假
参考代码:
- bool IsAVL(BinaryTreeNode * pRoot, int & height)
- {
- if(pRoot == NULL)
- {
- height = 0;
- return true;
- }
- int heightLeft;
- bool resultLeft = IsAVL(pRoot->m_pLeft, heightLeft);
- int heightRight;
- bool resultRight = IsAVL(pRoot->m_pRight, heightRight);
- if(resultLeft && resultRight && abs(heightLeft - heightRight) <= 1)
- {
- height = max(heightLeft, heightRight) + 1;
- return true;
- }
- else
- {
- height = max(heightLeft, heightRight) + 1;
- return false;
- }
- }
10. 求二叉树的镜像
递归解法:
(1)如果二叉树为空,返回空
(2)如果二叉树不为空,求左子树和右子树的镜像,然后交换左子树和右子树
参考代码如下:
- BinaryTreeNode * Mirror(BinaryTreeNode * pRoot)
- {
- if(pRoot == NULL)
- return NULL;
- BinaryTreeNode * pLeft = Mirror(pRoot->m_pLeft);
- BinaryTreeNode * pRight = Mirror(pRoot->m_pRight);
-
- pRoot->m_pLeft = pRight;
- pRoot->m_pRight = pLeft;
- return pRoot;
- }
11. 求二叉树中两个节点的最低公共祖先节点
递归解法:
(1)如果两个节点分别在根节点的左子树和右子树,则返回根节点
(2)如果两个节点都在左子树,则递归处理左子树;如果两个节点都在右子树,则递归处理右子树
参考代码如下:
- bool FindNode(BinaryTreeNode * pRoot, BinaryTreeNode * pNode)
- {
- if(pRoot == NULL || pNode == NULL)
- return false;
-
- if(pRoot == pNode)
- return true;
-
- bool found = FindNode(pRoot->m_pLeft, pNode);
- if(!found)
- found = FindNode(pRoot->m_pRight, pNode);
-
- return found;
- }
-
- BinaryTreeNode * GetLastCommonParent(BinaryTreeNode * pRoot,
- BinaryTreeNode * pNode1,
- BinaryTreeNode * pNode2)
- {
- if(FindNode(pRoot->m_pLeft, pNode1))
- {
- if(FindNode(pRoot->m_pRight, pNode2))
- return pRoot;
- else
- return GetLastCommonParent(pRoot->m_pLeft, pNode1, pNode2);
- }
- else
- {
- if(FindNode(pRoot->m_pLeft, pNode2))
- return pRoot;
- else
- return GetLastCommonParent(pRoot->m_pRight, pNode1, pNode2);
- }
- }
递归解法效率很低,有很多重复的遍历,下面看一下非递归解法。
非递归解法:
先求从根节点到两个节点的路径,然后再比较对应路径的节点就行,最后一个相同的节点也就是他们在二叉树中的最低公共祖先节点
参考代码如下:
- bool GetNodePath(BinaryTreeNode * pRoot, BinaryTreeNode * pNode,
- list<BinaryTreeNode *> & path)
- {
- if(pRoot == pNode)
- {
- path.push_back(pRoot);
- return true;
- }
- if(pRoot == NULL)
- return false;
- path.push_back(pRoot);
- bool found = false;
- found = GetNodePath(pRoot->m_pLeft, pNode, path);
- if(!found)
- found = GetNodePath(pRoot->m_pRight, pNode, path);
- if(!found)
- path.pop_back();
- return found;
- }
- BinaryTreeNode * GetLastCommonParent(BinaryTreeNode * pRoot, BinaryTreeNode * pNode1, BinaryTreeNode * pNode2)
- {
- if(pRoot == NULL || pNode1 == NULL || pNode2 == NULL)
- return NULL;
- list<BinaryTreeNode*> path1;
- bool bResult1 = GetNodePath(pRoot, pNode1, path1);
- list<BinaryTreeNode*> path2;
- bool bResult2 = GetNodePath(pRoot, pNode2, path2);
- if(!bResult1 || !bResult2)
- return NULL;
- BinaryTreeNode * pLast = NULL;
- list<BinaryTreeNode*>::const_iterator iter1 = path1.begin();
- list<BinaryTreeNode*>::const_iterator iter2 = path2.begin();
- while(iter1 != path1.end() && iter2 != path2.end())
- {
- if(*iter1 == *iter2)
- pLast = *iter1;
- else
- break;
- iter1++;
- iter2++;
- }
- return pLast;
- }
在上述算法的基础上稍加变化即可求二叉树中任意两个节点的距离了。
12. 求二叉树中节点的最大距离
即二叉树中相距最远的两个节点之间的距离。
递归解法:
(1)如果二叉树为空,返回0,同时记录左子树和右子树的深度,都为0
(2)如果二叉树不为空,最大距离要么是左子树中的最大距离,要么是右子树中的最大距离,要么是左子树节点中到根节点的最大距离+右子树节点中到根节点的最大距离,同时记录左子树和右子树节点中到根节点的最大距离。
参考代码如下:
- int GetMaxDistance(BinaryTreeNode * pRoot, int & maxLeft, int & maxRight)
- {
-
-
- if(pRoot == NULL)
- {
- maxLeft = 0;
- maxRight = 0;
- return 0;
- }
- int maxLL, maxLR, maxRL, maxRR;
- int maxDistLeft, maxDistRight;
- if(pRoot->m_pLeft != NULL)
- {
- maxDistLeft = GetMaxDistance(pRoot->m_pLeft, maxLL, maxLR);
- maxLeft = max(maxLL, maxLR) + 1;
- }
- else
- {
- maxDistLeft = 0;
- maxLeft = 0;
- }
- if(pRoot->m_pRight != NULL)
- {
- maxDistRight = GetMaxDistance(pRoot->m_pRight, maxRL, maxRR);
- maxRight = max(maxRL, maxRR) + 1;
- }
- else
- {
- maxDistRight = 0;
- maxRight = 0;
- }
- return max(max(maxDistLeft, maxDistRight), maxLeft+maxRight);
- }
13. 由前序遍历序列和中序遍历序列重建二叉树
二叉树前序遍历序列中,第一个元素总是树的根节点的值。中序遍历序列中,左子树的节点的值位于根节点的值的左边,右子树的节点的值位
于根节点的值的右边。
递归解法:
(1)如果前序遍历为空或中序遍历为空或节点个数小于等于0,返回NULL。
(2)创建根节点。
前序遍历的第一个数据就是根节点的数据,在中序遍历中找到根节点的位置,可分别得知左子树和右子树的前序和中序遍
历序列,重建左右子树。
题目描述:
-
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并输出它的后序遍历序列。
输入:
-
输入可能包含多个测试样例,对于每个测试案例,
输入的第一行为一个整数n(1<=n<=1000):代表二叉树的节点个数。
输入的第二行包括n个整数(其中每个元素a的范围为(1<=a<=1000)):代表二叉树的前序遍历序列。
输入的第三行包括n个整数(其中每个元素a的范围为(1<=a<=1000)):代表二叉树的中序遍历序列。
输出:
-
对应每个测试案例,输出一行:
如果题目中所给的前序和中序遍历序列能构成一棵二叉树,则输出n个整数,代表二叉树的后序遍历序列,每个元素后面都有空格。
如果题目中所给的前序和中序遍历序列不能构成一棵二叉树,则输出”No”。
样例输入:
-
81 2 4 7 3 5 6 84 7 2 1 5 3 8 681 2 4 7 3 5 6 84 1 2 7 5 3 8 6
样例输出:
-
7 4 2 5 8 6 3 1 No
采用递归的方式重构二叉树,关键是要考虑到一些特殊情况,比如:只有根节点的二叉树、只有左子树或只有右子树的二叉树以及二叉树根节点为NULL、前序中序序列不匹配导致不能重构二叉树等。
AC代码如下(一直在如何实现判断能否重构二叉树的地方徘徊,在九度论坛里大致看了下,借鉴了下各位前辈的思路:定义一个全局bool变量,用来跟踪判断能够重构):
- #include<stdio.h>
- #include<stdlib.h>
-
- typedef int ElemType;
- typedef struct BTNode
- {
- ElemType data;
- struct BTNode *left;
- struct BTNode *right;
- }BTNode,*BTree;
-
- bool CanReBuild;
-
-
-
-
- void RebuildBinaryTree(BTree *ppTree,int *pre,int *inv,int len)
- {
- if(pre==NULL || inv==NULL)
- {
- CanReBuild = false;
- return ;
- }
-
-
- if(len < 1)
- return ;
-
-
- int i;
- for(i=0;i<len;i++)
- if(pre[0] == inv[i])
- break;
-
- if(i >= len)
- {
- CanReBuild = false;
- return ;
- }
-
-
- *ppTree = (BTree)malloc(sizeof(BTNode));
- if(*ppTree == NULL)
- exit(EXIT_FAILURE);
- (*ppTree)->data = pre[0];
- (*ppTree)->left = NULL;
- (*ppTree)->right = NULL;
-
-
- RebuildBinaryTree(&(*ppTree)->left,pre+1,inv,i);
- RebuildBinaryTree(&(*ppTree)->right,pre+i+1,inv+i+1,len-i-1);
- }
-
- void BehTraverse(BTree pTree)
- {
- if(pTree != NULL)
- {
- if(pTree->left != NULL)
- BehTraverse(pTree->left);
- if(pTree->right != NULL)
- BehTraverse(pTree->right);
- printf("%d ",pTree->data);
- }
- }
-
- void DestroyTree(BTree pTree)
- {
- if(pTree)
- {
- if(pTree->left)
- DestroyTree(pTree->left);
- if(pTree->right)
- DestroyTree(pTree->right);
- free(pTree);
- pTree = NULL;
- }
- }
- int main()
- {
- int len;
- BTree pTree = NULL;
- while(scanf("%d",&len) != EOF)
- {
- int *pre = (int *)malloc(len*sizeof(int));
- int *inv = (int *)malloc(len*sizeof(int));
- if(pre == NULL || inv == NULL)
- exit(EXIT_FAILURE);
-
- int i;
- for(i=0;i<len;i++)
- scanf("%d",pre+i);
- for(i=0;i<len;i++)
- scanf("%d",inv+i);
-
- CanReBuild = true;
- RebuildBinaryTree(&pTree,pre,inv,len);
- if(CanReBuild)
- {
- BehTraverse(pTree);
- printf("\n");
- DestroyTree(pTree);
- }
- else
- printf("No\n");
-
- free(pre);
- free(inv);
- pre = NULL;
- inv = NULL;
- }
- return 0;
- }
相同原理的代码
#include <iostream>
#include<queue>
using namespace std;
class BinTree;
class BinTreeNode{
private:
int data;
BinTreeNode *left;
BinTreeNode *right;
public:
friend class BinTree;
BinTreeNode():data(int()),left(NULL),right(NULL){}
BinTreeNode(int temp,BinTreeNode *leftChild=NULL,BinTreeNode*rightChild=NULL):data(temp),left(leftChild),right(rightChild){}
};
class BinTree{
public:
void CreateBinTree(BinTreeNode *&t,const int *VLR,const int *LVR,int n)//根据前序中序遍历构造树
{
int k = 0;
if(n == 0){ t = NULL; return ;}
while(LVR[k] != VLR[0])
k++;
t = new BinTreeNode(LVR[k]);
CreateBinTree(t->left,VLR+1,LVR,k);
CreateBinTree(t->right,VLR+k+1,LVR+k+1,n-k-1);
}
void LevelOrder(BinTreeNode* t)const//分层遍历
{
if(t!=NULL)
{
queue<BinTreeNode* > Q;
BinTreeNode* p;
Q.push(t);
while(!Q.empty())
{
p = Q.front();//指向首部
Q.pop();//移除队列首部元素
cout<<p->data<<" ";
if(p->left)
Q.push(p->left);//往队列底部添加元素
if(p->right)
Q.push(p->right);
}
}
}
void CreateBinTree(const int* VLR,const int* LVR,int n)//利用前序和中序来构造出原来的树
{
CreateBinTree(root,VLR,LVR,n);
}
void LevelOrder()const//层次遍历
{ LevelOrder(root);}
private:
BinTreeNode *root;
};
int main()
{
int n;
cin>>n;
int *vlr = new int[n];
int *lvr = new int[n];
for(int i=0; i<n; i++)
cin>>vlr[i];
for(int i=0; i<n; i++)
cin>>lvr[i];
BinTree tree;
tree.CreateBinTree(vlr,lvr,n);
tree.LevelOrder();
return 0;
}
- BinaryTreeNode * RebuildBinaryTree(int* pPreOrder, int* pInOrder, int nodeNum)
- {
- if(pPreOrder == NULL || pInOrder == NULL || nodeNum <= 0)
- return NULL;
- BinaryTreeNode * pRoot = new BinaryTreeNode;
-
- pRoot->m_nValue = pPreOrder[0];
- pRoot->m_pLeft = NULL;
- pRoot->m_pRight = NULL;
-
- int rootPositionInOrder = -1;
- for(int i = 0; i < nodeNum; i++)
- if(pInOrder[i] == pRoot->m_nValue)
- {
- rootPositionInOrder = i;
- break;
- }
- if(rootPositionInOrder == -1)
- {
- throw std::exception("Invalid input.");
- }
-
- int nodeNumLeft = rootPositionInOrder;
- int * pPreOrderLeft = pPreOrder + 1;
- int * pInOrderLeft = pInOrder;
- pRoot->m_pLeft = RebuildBinaryTree(pPreOrderLeft, pInOrderLeft, nodeNumLeft);
-
- int nodeNumRight = nodeNum - nodeNumLeft - 1;
- int * pPreOrderRight = pPreOrder + 1 + nodeNumLeft;
- int * pInOrderRight = pInOrder + nodeNumLeft + 1;
- pRoot->m_pRight = RebuildBinaryTree(pPreOrderRight, pInOrderRight, nodeNumRight);
- return pRoot;
- }
同样,有中序遍历序列和后序遍历序列,类似的方法可重建二叉树,但前序遍历序列和后序遍历序列不同恢复一棵二叉树,证明略。
14.判断二叉树是不是完全二叉树
若设二叉树的深度为h,除第 h 层外,其它各层 (1~h-1) 的结点数都达到最大个数,第 h 层所有的结点都连续集中在最左边,这就是完全
二叉树。
有如下算法,按层次(从上到下,从左到右)遍历二叉树,当遇到一个节点的左子树为空时,则该节点右子树必须为空,且后面遍历的节点左
右子树都必须为空,否则不是完全二叉树。
- bool IsCompleteBinaryTree(BinaryTreeNode * pRoot)
- {
- if(pRoot == NULL)
- return false;
- queue<BinaryTreeNode *> q;
- q.push(pRoot);
- bool mustHaveNoChild = false;
- bool result = true;
- while(!q.empty())
- {
- BinaryTreeNode * pNode = q.front();
- q.pop();
- if(mustHaveNoChild)
- {
- if(pNode->m_pLeft != NULL || pNode->m_pRight != NULL)
- {
- result = false;
- break;
- }
- }
- else
- {
- if(pNode->m_pLeft != NULL && pNode->m_pRight != NULL)
- {
- q.push(pNode->m_pLeft);
- q.push(pNode->m_pRight);
- }
- else if(pNode->m_pLeft != NULL && pNode->m_pRight == NULL)
- {
- mustHaveNoChild = true;
- q.push(pNode->m_pLeft);
- }
- else if(pNode->m_pLeft == NULL && pNode->m_pRight != NULL)
- {
- result = false;
- break;
- }
- else
- {
- mustHaveNoChild = true;
- }
- }
- }
- return result;
- }