LeetCode-103-mid-二叉树的锯齿形层序遍历(dfs:深度优先搜索、搜索模板)

题目描述

LeetCode-103-mid-二叉树的锯齿形层序遍历

分析

对于层序遍历,不能够使用递归的方式,一般直接进行搜索。

类似题目:二叉树的层序遍历,使用队列存储每层要访问的对象。

本题在此之上要求相邻两层的遍历方向相反。对于第层的最后一个遍历的元素,其子节点优先被遍历,是和栈后进先出的特点相符合的。

解题思路

使用栈来存储每层待访问的节点指针。

搜索具有很明确的代码模板。

搜索模板

INIT:
    q : 线性表
    q.push(root)
ITERATION:
    while !q.empty():
        curr = q.front()
        q.pop()
        
        visit( curr ) // 遍历
        
        neighbors = get_neighbors( curr ) // 与curr相邻的待访问的节点
        q.push( neighbors )
 

算法

INPUTS: root
OUTPUTS: layers

ans = []
s_layers.push( root )
depth = 0

while !s_layers.empty() :
    next_layers = []
    layer_visit = []
    while !s_layers.empty():
        curr = s_layers.front()
        s_layers.pop()
        
        layer_visit.push( curr )
        
        if depth%2 :
            if curr->left != NULL :
                next_layer.push( curr->left )
            if curr->right != NULL :
                next_layer.push( curr->right )
        else :
            if curr->right != NULL :
                next_layer.push( curr->right )
            if curr->left != NULL :
                next_layer.push( curr->left )
    
    // BEGIN NEXT LAYER,维护循环不变量
    s_layers = next_layers
    ans.push( layer_visit )
    ++ depth
        

数据结构

这道题主要使用栈来进行遍历顺序上的模拟。

使用

复杂度分析

时间和空间复杂度都是

代码实现

class Solution {
public:
    vector> zigzagLevelOrder(TreeNode* root) {
        vector> ans ;
        if ( root == NULL ){
            return ans ;
        }
        
        stack s ;
        s.push( root ) ;
        int depth = 0 ;
        while ( !s.empty() ){
            // 遍历一层
            vector tmp ;
            stack nextS ;

            while ( !s.empty() ){
                TreeNode* curr = s.top() ;
                s.pop() ;

                tmp.push_back( curr->val ) ;

                if ( depth % 2 ){
                    if ( curr->right != NULL ){
                        nextS.push( curr->right ) ;
                    }
                    if ( curr->left != NULL ){
                        nextS.push( curr->left ) ;
                    }
                }else{
                    if ( curr->left != NULL ){
                        nextS.push( curr->left ) ;
                    }
                    if ( curr->right != NULL ){
                        nextS.push( curr->right ) ;
                    }
                }
            }
            ans.push_back( tmp ) ;
            s = nextS ; //可以么
            ++ depth ;
        }
        return ans ;
    }
};

相关问题

  • 二叉树的层序遍历

  • 二叉树的层序遍历


PS.

  1. 相比较于其他已有的leetcode刷题笔记,我希望能够提供相关的解题分析和部分相关问题的链接,使得大家能获得一个较好的分析与相关问题的比较。

  2. 偶尔会进行类似题目的总结工作,有需要的读者可以对这份工作进行关注。

  3. 如果你发现有相关的错误或者表述不当不清晰的地方,请进行评论,我会尽快进行核对勘正。

你可能感兴趣的:(LeetCode-103-mid-二叉树的锯齿形层序遍历(dfs:深度优先搜索、搜索模板))