其他分享
首页 > 其他分享> > LeetCode 589. N-ary Tree Preorder Traversal

LeetCode 589. N-ary Tree Preorder Traversal

作者:互联网

原题链接在这里:https://leetcode.com/problems/n-ary-tree-preorder-traversal/

题目:

Given the root of an n-ary tree, return the preorder traversal of its nodes' values.

Nary-Tree input serialization is represented in their level order traversal. Each group of children is separated by the null value (See examples)

Example 1:

Input: root = [1,null,3,2,4,null,5,6]
Output: [1,3,5,6,2,4]

Example 2:

Input: 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]
Output: [1,2,3,6,7,11,14,4,8,12,5,9,13,10]

Constraints:

Follow up: Recursive solution is trivial, could you do it iteratively?

题解:

Could use top down recursion.

Time Complexity: O(n). n is the number of nodes in the tree.

Space: O(logn). regardless res. stack space.

AC Java:

 1 /*
 2 // Definition for a Node.
 3 class Node {
 4     public int val;
 5     public List<Node> children;
 6 
 7     public Node() {}
 8 
 9     public Node(int _val) {
10         val = _val;
11     }
12 
13     public Node(int _val, List<Node> _children) {
14         val = _val;
15         children = _children;
16     }
17 };
18 */
19 
20 class Solution {
21     public List<Integer> preorder(Node root) {
22         List<Integer> res = new ArrayList<>();
23         if(root == null){
24             return res;
25         }
26         
27         dfs(root, res);
28         return res;
29     }
30     
31     private void dfs(Node root, List<Integer> res){
32         if(root == null){
33             return;
34         }
35         
36         res.add(root.val);
37         for(Node next : root.children){
38             dfs(next, res);
39         }
40     }
41 }

Could also do it iteratively. It is the method 2 in Binary Tree Preorder Traversal.

Time Complexity: O(n).

Space: O(logn). stack space.

AC Java:

 1 /*
 2 // Definition for a Node.
 3 class Node {
 4     public int val;
 5     public List<Node> children;
 6 
 7     public Node() {}
 8 
 9     public Node(int _val) {
10         val = _val;
11     }
12 
13     public Node(int _val, List<Node> _children) {
14         val = _val;
15         children = _children;
16     }
17 };
18 */
19 
20 class Solution {
21     public List<Integer> preorder(Node root) {
22         List<Integer> res = new ArrayList<>();
23         if(root == null){
24             return res;
25         }
26         
27         Stack<Node> stk = new Stack<>();
28         stk.push(root);
29         while(!stk.isEmpty()){
30             Node cur = stk.pop();
31             res.add(cur.val);
32             for(int i = cur.children.size() - 1; i >= 0; i--){
33                 if(cur.children.get(i) != null){
34                     stk.push(cur.children.get(i));
35                 }    
36             }
37         }
38         
39         return res;
40     }
41 }

 

标签:Node,Preorder,val,ary,Tree,res,null,root,children
来源: https://www.cnblogs.com/Dylan-Java-NYC/p/16414933.html