-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path144_BinaryTreePreorderTraversal.java
More file actions
73 lines (67 loc) · 1.8 KB
/
144_BinaryTreePreorderTraversal.java
File metadata and controls
73 lines (67 loc) · 1.8 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
/*
* Given a binary tree, return the preorder traversal of its nodes' values.
* For example:
* Given binary tree {1,#,2,3},
* 1
* \
* 2
* /
* 3
* return [1,2,3].
* Note: Recursive solution is trivial, could you do it iteratively?
*/
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
//recursive
public class Solution {
ArrayList<Integer> res = new ArrayList<Integer>();
public List<Integer> preorderTraversal(TreeNode root) {
traverse(root);
return res;
}
private void traverse(TreeNode node) {
if(node == null) return;
res.add(node.val);
traverse(node.left);
traverse(node.right);
}
}
//iterative 如果是空root怎么计算呢
public class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while(!stack.isEmpty()) {
TreeNode temp = stack.pop();
if(temp != null) {
res.add(temp.val);
stack.push(temp.right);
stack.push(temp.left);
}
}
return res;
}
}
//iterative2
public ArrayList<Integer> preorderTraversal(TreeNode root) {
ArrayList<Integer> res = new ArrayList<Integer>();
Stack<TreeNode> stack = new Stack<TreeNode>();
while(!stack.isEmpty() || root != null) {
if(root != null) {
res.add(root.val);
stack.push(root.right);
root = root.left;
}
else
root = stack.pop();
}
return res;
}