
589. N叉樹的前序遍歷
給定一個 N 叉樹,返回其節點值的前序遍歷。
例如,給定一個 3叉樹
:

返回其前序遍歷: [1,3,5,6,2,4]
。
說明: 遞歸法很簡單,你可以使用迭代法完成此題嗎?
題解:
既然是樹的遍歷,那么一共就是兩種思路,即深度優先搜索遍歷和廣度優先搜索遍歷。其中遞歸法就是深度優先搜索遍歷的思想,我們從左到右依次遍歷N叉樹的每個節點。
具體代碼如下:
/*
// Definition for a Node.
class Node {public int val;public List<Node> children;public Node() {}public Node(int _val) {val = _val;}public Node(int _val, List<Node> _children) {val = _val;children = _children;}
};
*/class Solution {List<Integer> result = new ArrayList<>();public List<Integer> preorder(Node root) {if (root == null) return result;result.add(root.val);//遍歷當前節點的所有孩子節點for (int i = 0; i < root.children.size(); i++) {preorder(root.children.get(i));}return result;}
}
題目要求思考迭代方法來遍歷,能想到的就是借鑒廣度優先搜索遍歷的方法。創建一個棧,每次棧頂彈出的元素就是前序遍歷的順序。具體來說,逆序每個節點的孩子節點,然后依次入棧,每次pop()
棧頂的元素,就是前序遍歷的順序。
具體代碼如下:
/*
// Definition for a Node.
class Node {public int val;public List<Node> children;public Node() {}public Node(int _val) {val = _val;}public Node(int _val, List<Node> _children) {val = _val;children = _children;}
};
*/class Solution {List<Integer> result = new ArrayList<>();public List<Integer> preorder(Node root) {if (root == null) return result;Deque<Node> deque = new ArrayDeque<>();deque.add(root);while (!deque.isEmpty()) {Node node = deque.pop();result.add(node.val);//每次逆序入棧當前節點的所有孩子節點Collections.reverse(node.children);for (Node child : node.children) {deque.push(child);}}return result;}
}