2012/3/13 二叉树的层序遍历（leetcode 102）

这是一道队列相关的例题

参考b站“懒猫老师”视频

伪代码

• 队列Q初始化
• 如果二叉树非空，将根指针入队
• 循环直到队列Q为空（循环的退出条件）
  - q=队列Q的队头元素出队
  - 访问节点q的数据域（这一步输出了）
  - 若节点q存在左孩子，则将左孩子指针入队
  - 若节点q存在右孩子，则将右孩子指针入队
  这个方法每次弹出一个值 并压入它的所有子节点
  完整代码参考

https://blog.csdn.net/ZM970307/article/details/79934150

#include <iostream>
#include <vector>
#include <queue>
using namespace std;

typedef char TElemType;
typedef enum{ Link, Thread } PointerTag;
typedef struct BiTNode
{
	TElemType data;
	struct BiTNode* lChild;
	struct BiTNode* rChild;
	PointerTag lTag;
	PointerTag rTag;

}BiTNode, *BiTree;

unsigned int  TElemIndex = 0;   //用于从数组创建树

/*从数组创建树*/
void createBiTreeByArray(BiTree *T, vector<TElemType> list)
{
	if (TElemIndex + 1 <= list.size())
	{
		TElemType temp = list[TElemIndex];
		if (temp == '#')
		{
			*T = NULL;
			TElemIndex++;       /*index必须使用全局变量++，
								*而不能作为函数参数++(函数局部变量index在内层递归index++不会改变外层的index,退出内层递归时index会变回去)*/
		}
		else
		{
			*T = (BiTNode *)malloc(sizeof(BiTNode));
			if (!(*T))
			{
				perror("fail: create-malloc");
				return;
			}
			(*T)->data = temp;
			(*T)->lChild = NULL;
			(*T)->rChild = NULL;
			(*T)->lTag = Link;
			(*T)->rTag = Link;
			TElemIndex++;
			createBiTreeByArray(&(*T)->lChild, list);
			createBiTreeByArray(&(*T)->rChild, list);
		}
	}
}

/*二叉树层序遍历*/
void levelTraverse(BiTree T)
{
	queue<BiTree> q;        //定义一个队列，数据类型是二叉树结点指针，不要仅是TElemType！不然无法遍历
	if (T)
	{
		q.push(T);
	}
	while (!q.empty())
	{
		BiTree front = q.front();       //取队列最前面的结点
		cout << front->data;
		q.pop();        //删除最前面的结点

		if (front->lTag == Link && front->lChild)       //判断最前面的左结点是否为空，不是则放入队列
		{
			q.push(front->lChild);
		}
		if (front->rTag == Link && front->rChild)   //判断最前面的右结点是否为空，不是则放入队列
		{
			q.push(front->rChild);
		}
	}
}

int main()
{
	vector<TElemType> TElemList = { 'A', 'B', 'D', 'H', '#', 'K', '#', '#', '#', 'E', '#', '#', 'C', 'F', 'I', '#', '#', '#', 'G', '#', 'J', '#', '#' };
	BiTree root = NULL;

	createBiTreeByArray(&root, TElemList);
	cout << "create ok." << endl;

	cout << "层序遍历：";
	levelTraverse(root);
	cout << endl;

	system("pause");
	return 0;
}

leetcode官方解答改良了这里 每次弹出si个数 并压入它们的子节点 他的返回值是一个二维数组

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector <vector <int>> ret;
        if (!root) {
            return ret;
        }

        queue <TreeNode*> q;
        q.push(root);
        while (!q.empty()) {
            int currentLevelSize = q.size();
            ret.push_back(vector <int> ());
            for (int i = 1; i <= currentLevelSize; ++i) {
                auto node = q.front(); q.pop();
                ret.back().push_back(node->val);
                if (node->left) q.push(node->left);
                if (node->right) q.push(node->right);
            }
        }
        
        return ret;
    }
};//这个的输出格式是二维数组 好像没办法用最基础的bfs？

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal/solution/er-cha-shu-de-ceng-xu-bian-li-by-leetcode-solution/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

看到了一个写的更清晰的代码

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
    vector<vector<int>> res;
    if(!root) return res;
    queue<TreeNode*> qu;
    qu.push(root);
    while(!qu.empty())
    {
        vector<int> tmp;//这个里面放的是一层的节点
        int len=qu.size();
        for(int i=0;i<len;i++){
            TreeNode* node=qu.front();
            qu.pop();
            tmp.push_back(node->val);
            if(node->left) qu.push(node->left);
            if(node->right) qu.push(node->right);
        }
    res.push_back(tmp);//把每一层的节点压进去
    }
    return res;
    }
};