-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathq4p9.java
More file actions
99 lines (84 loc) · 2.82 KB
/
Copy pathq4p9.java
File metadata and controls
99 lines (84 loc) · 2.82 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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
import java.util.*;
import java.lang.*;
public class q4p9{
public static void main(String[] args){
int[] arr = {20,60,10,25,70};
Node root = new Node(50);
for(int n : arr){
buildTree(root, n);
}
inNode(root);
System.out.println("=============");
ArrayList<LinkedList<Integer>> ans = allSequences(root);
System.out.println(ans);
}
public static void inNode(Node root){
if(root == null) return;
inNode(root.left);
System.out.println(root.value);
inNode(root.right);
}
public static void buildTree(Node root, int value){
if(root == null)
return;
if(value < root.value){
if(root.left == null){
root.left = new Node(value);
}else{
buildTree(root.left, value);
}
}else{
if(root.right == null){
root.right = new Node(value);
}else{
buildTree(root.right, value);
}
}
}
public static ArrayList<LinkedList<Integer>> allSequences(Node node){
ArrayList<LinkedList<Integer>> result = new ArrayList<>();
if(node == null){
result.add(new LinkedList<Integer>());
return result;
}
LinkedList<Integer> prefix = new LinkedList<>();
prefix.add(node.value);
ArrayList<LinkedList<Integer>> leftSeq = allSequences(node.left);
ArrayList<LinkedList<Integer>> rightSeq = allSequences(node.right);
for(LinkedList<Integer> left: leftSeq){
for(LinkedList<Integer> right: rightSeq){
ArrayList<LinkedList<Integer>> weaved = new ArrayList<>();
weaveLists(left, right, weaved, prefix);
result.addAll(weaved);
}
}
return result;
}
public static void weaveLists(LinkedList<Integer> first, LinkedList<Integer> second, ArrayList<LinkedList<Integer>> results, LinkedList<Integer> prefix){
if(first.size() == 0 || second.size() == 0){
LinkedList<Integer> result = (LinkedList<Integer>) prefix.clone();
result.addAll(first);
result.addAll(second);
results.add(result);
return;
}
int headFirst = (int) first.removeFirst();
prefix.addLast(headFirst);
weaveLists(first, second, results, prefix);
prefix.removeLast();
first.addFirst(headFirst);
int headSecond = (int) second.removeFirst();
prefix.addLast(headSecond);
weaveLists(first, second, results, prefix);
prefix.removeLast();
second.addFirst(headSecond);
}
}
class Node{
int value;
Node left = null;
Node right = null;
public Node(int value){
this.value = value;
}
}