剑指offer——59/66按之字形顺序打印二叉树

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

思路:层次遍历,但是偶数层的需要反着添加!

class Solution {
public:
    void reverse(vector<int>&vec)
    {
        for(int i=0;i<vec.size()/2;i++)
        {
            swap(vec[i],vec[vec.size()-1-i]);
        }
    }
    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int>>res;
        vector<int>vec;
        queue<TreeNode *>que;
        if(pRoot)
        {
            que.push(pRoot);
            int t=0;
            while(!que.empty())
            {
                int n=que.size();
                for(int i=0;i<n;i++)
                {
                    TreeNode *node=que.front();
                    que.pop();
                    vec.push_back(node->val);
                    if(node->left)
                        que.push(node->left);
                    if(node->right)
                        que.push(node->right);
                }
                if(t%2==0)
                    res.push_back(vec);
                else
                {
                    reverse(vec);
                    res.push_back(vec);
                }
                vec.clear();
                t++;
            }
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_17141957/article/details/81040221
今日推荐