二叉树_题目25:589. N 叉树的前序遍历

41 阅读2分钟

题目25:589. N 叉树的前序遍历

589. N 叉树的前序遍历

同类题目

前置知识

题目描述

给定一个 n 叉树的根节点 root ,返回 其节点值的 前序遍历

n 叉树 在输入中按层序遍历进行序列化表示,每组子节点由空值 null 分隔(请参见示例)。

示例 1:

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

示例 2:

 输入: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]

提示:

  • 节点总数在范围 [0, 104]
  • 0 <= Node.val <= 104
  • n 叉树的高度小于或等于 1000

进阶: 递归法很简单,你可以使用迭代法完成此题吗?

思路分析

递归

  • 跟二叉树的前序遍历相似
  • 不同点在于这里不仅仅有两个孩子,而是有多个孩子,按照从左到右的顺序压入即可。

程序代码

方法一

 // https://leetcode.cn/problems/n-ary-tree-preorder-traversal/description/
 // 589. N 叉树的前序遍历
 #include<iostream>
 #include<vector>
 #include<stack>
 using namespace std;
 ​
 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 {
 private:
     vector<int> res;
 public:
     vector<int> travel(Node* root)
     {
         if(root == nullptr)
         {
             return res;
         }
 ​
         res.push_back(root->val);
         // 按照顺序压入即可
         for(int i=0;i<root->children.size();i++)
         {
             travel(root->children[i]);
         }
         return res;
     }
     vector<int> preorder(Node* root) {
         res = travel(root);
         return res;
     }
 };
 ​
 ​

复杂度分析

时间复杂度O(N),一共有 N个节点,每个节点都需要遍历一次。

空间复杂度O(N) ,主要是数组的开销。

思路分析

迭代

按照从右到左的顺序压入栈,因为栈是先进后出。

程序代码

 // 使用迭代的方法
 class Solution {
 public:
     vector<int> preorder(Node* root) {
         vector<int> res;
         if(root == nullptr)
         {
             return res;
         }
 ​
         stack<Node*> sta;
         sta.push(root);
         while(!sta.empty())
         {
             Node *cur = sta.top();
             res.push_back(cur->val);
             sta.pop();
             // 注意这里要反序,因为出栈是顺序是 后入先出
             for(int i=cur->children.size()-1;i>=0;i--)
             {
                 sta.push(cur->children[i]);
             }
         }
    
         return res;
     }
 };

题目感悟

更多相关知识:github.com/pingguo1987…