题解 | #按之字形顺序打印二叉树#

按之字形顺序打印二叉树

http://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0

在经典的层序遍历的基础上,对层进行就判断,偶数层就进行翻转;奇数层则保持不变。

/*
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> > ans;
        if (!pRoot) return ans;   // 空指针,直接返回
        queue<TreeNode*> que;
        que.push(pRoot);
        int c = 0;     // 记录奇偶层
        while(!que.empty()) {  // 队列不为空
            int n = que.size();
            vector<int> tmp;
            while (n--) {   // 当前层遍历
                tmp.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();
            }
            c++;
            if (c%2 == 0) {
                reverse(tmp.begin(), tmp.end());  // 偶数层,则反转
            }
            ans.emplace_back(tmp);
        }
        return ans;
    }

};
全部评论

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务