剑指offer 按之字形顺序打印二叉树

题目描述

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

 

 

思路：

1、有了之前层次遍历一行一行输出的经验，我们可以直接用一个变量记录行数，如果是奇数行，就将队列中的元素按顺序所有保存下来，如果是偶数行，就顺序保存到一个临时vector中，再逆序保存下来。

/*
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) {
        vector<vector<int> > res;
        vector<int> temp;
        int odd = 0; //0表示奇数行，1表示偶数行
        queue<TreeNode*> q;
        if (pRoot == NULL)
            return res;
        q.push(pRoot);
        int current = 1, next = 0; //current记录当前打印行的元素数，next表示下一行的元素个数。
        while (!q.empty()) {
            temp.clear();
            for (int i = 0; i < current; i++) {
                TreeNode* pNode = q.front();
                q.pop();
                temp.push_back(pNode->val);
                if (pNode->left != NULL) {
                    q.push(pNode->left);
                    next++;
                }
                if (pNode->right != NULL) {
                    q.push(pNode->right);
                    next++;
                }
            }
            //如果是奇数行，就要将保存下来的逆序保存
            if (odd == 1) {
                for (int i = 0; i < (current >> 1); i++) {
                    swap(temp[i], temp[current - 1 - i]);
                }
            }
            res.push_back(temp);
            odd = 1 - odd;
            current = next;
            next = 0;
        }
        return res;
    }
}；

 

 

2、剑指offer思路

/*
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) {
        vector<vector<int> > res;
        vector<int> temp;
        stack<TreeNode*> level[2]; 
        int current = 0, next = 1; 
        if (pRoot == NULL)
            return res;
        level[current].push(pRoot);
        //如果pRoot == NULL, 那么压入栈之后，栈非空，而取元素的话，又会产生越界？
        while (!level[current].empty() || !level[next].empty()) {
            TreeNode* pNode = level[current].top();
            level[current].pop();
            temp.push_back(pNode->val);
            //current == 0表示当前在保存偶数行的数，将下一行的数保存到栈时，先将节点的左孩子放入栈中
            if (current == 0) {
                if (pNode->left != NULL)
                    level[next].push(pNode->left);
                if (pNode->right != NULL)
                    level[next].push(pNode->right);
            } else { //当前在保存奇数行的数据，先将节点的右孩子放入栈中
                if (pNode->right != NULL)
                    level[next].push(pNode->right);
                if (pNode->left != NULL)
                    level[next].push(pNode->left);
            }
            if (level[current].empty()) {
                res.push_back(temp);
                temp.clear();
                current = 1 - current;
                next = 1 - next;
            }
        }
        return res;
    }
};