-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathDFS.java
More file actions
58 lines (50 loc) · 1.54 KB
/
DFS.java
File metadata and controls
58 lines (50 loc) · 1.54 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
import java.util.List;
import java.util.ArrayList;
import java.util.Stack;
public class DFS {
public List<Integer> dfs(TreeNode root) {
if (root == null)
return new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
return dfs(stack);
}
public List<Integer> dfs(Stack<TreeNode> stack) {
List<Integer> list = new ArrayList<>();
while (!stack.isEmpty()) {
TreeNode node = stack.pop();
list.add(node.val);
if (node.right != null)
stack.push(node.right);
if (node.left != null)
stack.push(node.left);
}
return list;
}
public static void main(String[] args) {
DFS dfs = new DFS();
TreeNode root = new TreeNode(7);
TreeNode left = new TreeNode(4);
TreeNode right = new TreeNode(9);
root.left = left;
root.right = right;
TreeNode leftLeft = new TreeNode(2);
TreeNode leftRight = new TreeNode(3);
left.left = leftLeft;
left.right = leftRight;
TreeNode leftLeftLeft = new TreeNode(1);
leftLeft.left =leftLeftLeft;
TreeNode rightLeft = new TreeNode(8);
TreeNode rightRight = new TreeNode(10);
right.left = rightLeft;
right.right = rightRight;
List<Integer> traversal = dfs.dfs(root);
System.out.println(traversal);
}
}
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int val) { this.val = val; }
}