剑指offer 按之字形顺序打印二叉树（C++）

题目描述

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

解题思路

• 本题目就是二叉树层序遍历的变形，在程序主题中可以定义一个queue队列数据结构来方便二叉树的遍历。

• 本题目返回一个动态二维数组即vector<<vector>int >类型，思路可以是定义一个一维数组vector<int> tmpVec;，每次这个二维数组接收这个一维数组即可。

• 由于打印之字形的二叉树，即定义一个bool even = false;由于第一行是正序的，所以even的初始值定义为false。待遍历第二层结点时，需要将元素逆序，此时even = !even;即可进行进入元素逆序的部分。

• 每次用int num = que.size();类型变量记录当前队列中元素的个数，然后依次将结点值输入给tmpVec数组，逐一将每个对头结点出队的同时，左右子树进队，以此类推，直到队列中再没有元素为止。

代码实现

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public: // 二叉树层序遍历的变体
    vector<vector<int> > Print(TreeNode* pRoot) {
        if(!pRoot)
            return {};
        bool even = false;
        vector<vector<int> > vec; // 多借助其他辅助数据结构
        vector<int> tmpVec;
        queue<TreeNode*> que; 
        que.push(pRoot); // 此时根结点不空  根结点入队
        while(!que.empty())
        {
            int num = que.size(); // 当前队列中二叉树结点个数
            for(int i = 0; i < num; i++)
            {
                tmpVec.push_back(que.front()->val); // 对头结点值进入数组
                if(que.front()->left) // 结点左子树不空
                    que.push(que.front()->left); // 左子树进队
                if(que.front()->right) // 结点右子树不空
                    que.push(que.front()->right); // 右子树进队
                que.pop(); // 对头元素出队
            }
            if(even) // 保证每循环两次时逆转数组中的元素
                reverse(tmpVec.begin(), tmpVec.end());
            vec.push_back(tmpVec); // 一维数组进入二维数组
            tmpVec.clear(); // 清空数组 以便用于存储下一层的结点值
            even = !even; // bool变量值发生改变
        }
        return vec; // 返回二维数组
    }
};

下面附带二叉树的层序遍历算法：

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

typedef struct BITREENODE
{
	int val;
	struct BITREENODE* left;
	struct BITREENODE* right;
}BiTreeNode;

vector<int> BiTreeLevelOrder(BiTreeNode* pRoot)
{
	if(pRoot == NULL) // 二叉树为空 返回空
		return {};
	queue<BiTreeNode*> que;
	vector<int> vec;
	que.push(pRoot); // 根结点入队
	while(!que.empty()) // 如果队列不空
	{
		vec.push_back(que.front()->val); // 对头结点值存入数组
		if(que.front()->left) // 对头结点左孩子不空
			que.push(que.front()->left); //左孩子入队
		if(que.front()->right) // 对头结点右孩子不空
			que.push(que.front()->right); // 右孩子入队
		que.pop(); // 对头结点出队
	}
	return vec; // 返回二叉树的层序遍历序列
}

总结与反思

• 可以定义除了函数本身给出以外的数据结构，只要对解题有利即可。
• 要勇敢的去定义变量，只要对解题有利即可。
• 二分类判断不一定要定义计数变量加加再对二取余，直接定义bool变量即可，要充分的从C思维转换到C++思维上来。