summaryrefslogtreecommitdiff
path: root/src/main
diff options
context:
space:
mode:
Diffstat (limited to 'src/main')
-rw-r--r--src/main/Toposort.java94
1 files changed, 82 insertions, 12 deletions
diff --git a/src/main/Toposort.java b/src/main/Toposort.java
index ede04b4..17cebff 100644
--- a/src/main/Toposort.java
+++ b/src/main/Toposort.java
@@ -1,42 +1,45 @@
import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;
+import java.util.Stack;
+import java.util.Arrays;
public class Toposort {
private static ArrayList<Integer> toposort(ArrayList<ArrayList<Integer>> adj) {
- int[] visited = new int[adj.size()];
- ArrayList<Integer> postv = new ArrayList<>(Collections.nCopies(adj.size(),0));
+ boolean[] visited = new boolean[adj.size()];
+ Integer[] postv = new Integer[adj.size()];
int[] clock = new int[1];
dfs(adj, visited, postv, clock);
- ArrayList<Integer> reverse_postv = (ArrayList<Integer>) postv.clone();
- Collections.sort(reverse_postv, Collections.reverseOrder());
+ Integer[] reverse_postv = postv.clone();
+ Arrays.sort(reverse_postv, Collections.reverseOrder());
- ArrayList<Integer> order = new ArrayList<Integer>();
+ ArrayList<Integer> order = new ArrayList<>();
for (int n : reverse_postv) {
- order.add(postv.indexOf(n));
+ order.add(Arrays.asList(postv).indexOf(n));
}
return order;
+
}
- private static void dfs(ArrayList<ArrayList<Integer>> adj, int[] visited, ArrayList<Integer> postv, int[] clock) {
+ private static void dfs(ArrayList<ArrayList<Integer>> adj, boolean[] visited, Integer[] postv, int[] clock) {
for (int v = 0; v < adj.size(); v++) {
- if (visited[v] == 0)
+ if (visited[v] == false)
explore(adj, v, visited, postv, clock);
}
}
- private static void explore(ArrayList<ArrayList<Integer>> adj, int v, int[] visited, ArrayList<Integer> postv, int[] clock) {
- visited[v] = 1;
+ private static void explore(ArrayList<ArrayList<Integer>> adj, int v, boolean[] visited, Integer[] postv, int[] clock) {
+ visited[v] = true;
clock[0] = clock[0] + 1;
for (int n : adj.get(v)) {
- if (visited[n] == 0)
+ if (visited[n] == false)
explore(adj, n, visited, postv, clock);
}
- postv.set(v, clock[0]);
+ postv[v] = clock[0];
clock[0] = clock[0] + 1;
}
@@ -60,5 +63,72 @@ public class Toposort {
System.out.print((x + 1) + " ");
}
}
+
+ static ArrayList<Integer> postOrder(ArrayList<ArrayList<Integer>> adj) {
+ boolean[] visited = new boolean[adj.size()];
+ ArrayList<Integer> postv = new ArrayList<>(Collections.nCopies(adj.size(), 0));
+ int[] clock = new int[1];
+ clock[0] = clock[0] + 1;
+ for (int v = 0; v < adj.size(); v++) {
+ exploreIterative(adj, v, visited, postv, clock);
+ }
+ return postv;
+ }
+
+ private static void exploreIterative(ArrayList<ArrayList<Integer>> adj, int v, boolean[] visited, ArrayList<Integer> postv, int[] clock) {
+ if (visited[v] == false) {
+ // prev[v] = clock[0]
+ boolean[] inStack = new boolean[adj.size()];
+ clock[0] = clock[0] + 1;
+ visited[v] = true;
+ inStack[v] = true;
+ Stack<Integer> vertexStack = new Stack<>();
+ vertexStack.push(v);
+
+ while (vertexStack.isEmpty() == false) {
+ int tempVertex = vertexStack.peek();
+// if (tempVertex != v) {
+// visited[tempVertex] = true;
+// // prev[v] = clock[0]
+// clock[0] = clock[0] + 1;
+// }
+ if (tempVertex != v && visited[tempVertex]) {
+ vertexStack.pop();
+ continue;
+ }
+ if (adj.get(tempVertex).size() == 0) {
+ if (tempVertex != v) {
+ // prev[tempVertex] = clock[0];
+ clock[0] = clock[0] + 1;
+ visited[tempVertex] = true;
+ }
+ postv.set(tempVertex, clock[0]);
+ clock[0] = clock[0] + 1;
+ vertexStack.pop();
+ } else {
+ boolean allEdgesFollowed = true;
+ for (int i : adj.get(tempVertex)) {
+ if (visited[i] == false) {
+// visited[i] = true;
+ // prev[i] = clock[0]
+ if (inStack[i] == false) {
+ inStack[i] = true;
+ vertexStack.push(i);
+ }
+ allEdgesFollowed = false;
+ }
+ }
+ if (allEdgesFollowed) {
+ visited[tempVertex] = true;
+ if (postv.get(tempVertex) == 0)
+ postv.set(tempVertex, clock[0]);
+ clock[0] = clock[0] + 1;
+ vertexStack.pop();
+ }
+ }
+ }
+ }
+ }
+
}