N叉树的前序遍历

想要知识进脑子

于 2024-02-18 23:53:04 发布

阅读量526

点赞数 13

分类专栏：算法题总结文章标签：深度优先算法

本文链接：https://blog.csdn.net/qq_45776114/article/details/136160845

版权

算法题总结专栏收录该内容

5 篇文章 0 订阅

订阅专栏

文章介绍了如何用递归和迭代两种方法实现N叉树的前序遍历，包括递归法中利用栈模拟子树遍历的过程以及迭代法中直接操作栈的思路。

摘要由CSDN通过智能技术生成

N叉树的前序遍历

题目描述

给定一个n叉树的根节点 root ,返回其节点值的前序遍历。n叉树在输中按层序遍历进行序列化表示，每组子节点由空值 null 分隔。

示例一

在这里插入图片描述

输入：root = [1,null,3,2,4,null,5,6]
输出：[1,3,5,6,2,4]

示例二

在这里插入图片描述

输入：root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出：[1,2,3,6,7,11,14,4,8,12,5,9,13,10]

方法一

树的遍历很容易想到 递归法.N叉树遍历可以类比于二叉树遍历.

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> children;

    Node() {}

    Node(int _val) {
        val = _val;
    }

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/

class Solution {
public:
    vector<int> preorder(Node* root) {
        vector<int> res;
        dfs(root, res);
        return res;
    }
    void dfs(Node* root, vector<int> & res) {
        if (root == nullptr) {
            return;
        }
        res.push_back(root->val);
        vector<Node*> children = root->children;
        if (children.size() > 0) { // 相比二叉树遍历子树的区别
            for (auto tmp : children) {
                dfs(tmp, res);
            }
        }
    }
};

方法二

迭代法，使用栈去模拟递归。

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> children;

    Node() {}

    Node(int _val) {
        val = _val;
    }

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/

class Solution {
public:
    vector<int> preorder(Node* root) {
        vector<int> res;
        if (root == nullptr){
            return res;
        }
        stack<Node*> st;
        st.push(root);
        while (!st.empty()){
            Node* tmp = st.top();
            st.pop();
            res.push_back(tmp->val);
            vector<Node*> children = tmp->children;
            for (int i = children.size() -1; i >= 0; i--) {//栈的特性先进后出，所以从右开始遍历
                st.push(children[i]);
            }
        }
        return res;
    }
};