【劍指offer】之字形打印二叉樹，C++實現

原創文章，轉載請注明出處！

本題牛客網地址

博客文章索引地址

博客文章中代碼的github地址

1.題目

      請實現一個函數按照之字形打印二叉樹，即第一行按照從左到右的順序打印，第二層按照從右至左的順序打印，第三行按照從左到右的順序打印，其他行以此類推。

例如：

輸入二叉樹

輸出順序

1

3 2

4 5 6 7

15 14 13 12 12 10 9 8

2.思路

      本題使用兩個棧作為輔助容器。打印某一層節點時，把下一層的子節點保存到棧內。如果當前打印的是奇數層，則先保存左子樹節點再保存右子樹節點到第一個棧內；如果當前打印的是偶數層，則先保存右子樹在保存左子樹節點到第二個棧內。

3.代碼

/*
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> > results;
        vector<int> rlt_temp;

        // 邊界條件
        if(pRoot == nullptr)
            return results;

        // 輔助容器
        stack<TreeNode*> stk[2]; // stk[0]是奇數層，stk[1]是偶數層
        int now = 0;
        int next = 1;
        TreeNode* temp=pRoot;

        // 根節點入棧
        stk[now].push(temp);

        // 遍歷兩個棧，當兩個棧均為空時，跳出循環
        while(!stk[now].empty() || !stk[next].empty()){
            // 存儲遍歷結果
            temp = stk[now].top();
            rlt_temp.push_back(temp->val);
            stk[now].pop();

            // 當前層是奇數或偶數
            if(now==0)
            {
                // 當前層是奇數時，左子樹先入棧，右子樹后入棧
                if(temp->left!=nullptr)
                    stk[next].push(temp->left);
                if(temp->right!=nullptr)
                    stk[next].push(temp->right);
            }
            else
            {
                // 當前層是偶數時，右子樹先入棧，左子樹后入棧
                if(temp->right!=nullptr)
                    stk[next].push(temp->right);
                if(temp->left!=nullptr)
                    stk[next].push(temp->left);
            }

            // 當前層為空時，打印下一層
            if(stk[now].empty())
            {
                results.push_back(rlt_temp);
                rlt_temp.clear();
                now=1-now;
                next = 1-next;
            }
        }
        return results;
    }
};