leetcode || 103、Binary Tree Zigzag Level Order Traversal

problem:

Given a binary tree, return the zigzag level order traversal of its nodes' values. (ie, from left to right, then right to left for the next level and alternate between).

For example:
Given binary tree {3,9,20,#,#,15,7},

    3
   / \
  9  20
    /  \
   15   7

return its zigzag level order traversal as:

[
  [3],
  [20,9],
  [15,7]
]

confused what "{1,#,2,3}" means? > read more on how binary tree is serialized on OJ.

Hide Tags
  Tree Breadth-first Search Stack
题意:Z字形层序遍历二叉树

thinking:

(1)层序遍历采用BFS广度优先搜索,借用队列queue

(2)要实现Z字形 层序遍历,要采用一个BOOL 值标记先访问左还是右孩子的顺序,还要借助两个queue,

其中一个用于输出VAL值,另外一个要借助stack实现翻转,保存访问下一层的孩子节点

code:

class Solution {
  private:
      vector<vector<int> > ret;
  public:
      vector<vector<int> > zigzagLevelOrder(TreeNode *root) {
          ret.clear();
          if(root==NULL)
              return ret;
          queue<TreeNode *> tmp_queue;
          tmp_queue.push(root);
          level_order(tmp_queue,false);
          return ret;
      }
  protected:
      void level_order(queue<TreeNode *> &queue1,bool flag)
      {
          if(queue1.empty())
              return;
          vector<int> array;
          queue<TreeNode *> queue0=queue1;
          queue<TreeNode *> queue2=queue_reverse(queue1);
          queue<TreeNode *> queue3;
          while(!queue0.empty()) //打印结点
          {

              TreeNode *tmp=queue0.front();
               queue0.pop();
              array.push_back(tmp->val);
          }
          while(!queue2.empty())   //访问孩子节点
          {
              TreeNode *tmp=queue2.front();
              queue2.pop();
              if(flag)   //flag规定先访问左孩子还是右孩子
              {
                if(tmp->left!=NULL)
                    queue3.push(tmp->left);
                if(tmp->right!=NULL)
                    queue3.push(tmp->right);
              }
              else
              {
                  if(tmp->right!=NULL)
                      queue3.push(tmp->right);
                  if(tmp->left!=NULL)
                      queue3.push(tmp->left);
              }
          }
          flag=!flag;
          ret.push_back(array);
          level_order(queue3,flag);  //递归访问下一层
      }
      queue<TreeNode *> queue_reverse(queue<TreeNode *> &_queue)   //借助stack实现queue翻转
      {
          stack<TreeNode *> _stack;
          queue<TreeNode *> ret_queue;
          while(!_queue.empty())
          {
              TreeNode *tmp=_queue.front();
              _stack.push(tmp);
              _queue.pop();
          }
          while(!_stack.empty())
          {
              TreeNode *tmp=_stack.top();
              ret_queue.push(tmp);
              _stack.pop();
          }
          return ret_queue;
      }
  };


你可能感兴趣的:(LeetCode,二叉树,Queue,bfs,层序遍历)