[数据结构与算法]【ACWing】45. 之字形打印二叉树

题目地址：

https://www.acwing.com/problem/content/43/

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

数据范围：
树中节点的数量$[0,1000]$。

代码如下：

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

struct TreeNode {
  int val;
  TreeNode *left;
  TreeNode *right;
  TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
  vector<vector<int>> printFromTopToBottom(TreeNode* root) {
    vector<vector<int>> res;
    if (!root) return res;
    bool rev = false;
    queue<TreeNode*> q;
    q.push(root);
    while (q.size()) {
      vector<int> row;
      for (int i = q.size(); i; i--) {
        auto t = q.front(); q.pop();
        row.push_back(t->val);
        if (t->left) q.push(t->left);
        if (t->right) q.push(t->right);
      }
      if (rev) reverse(row.begin(), row.end());
      res.push_back(row);
      rev = !rev;
    }

    return res;
  }
};

时空复杂度$O(n)$。

创作打卡挑战赛

赢取流量/现金/CSDN周边激励大奖