|
24 | 24 | * </p>
|
25 | 25 | */
|
26 | 26 |
|
27 |
| -public final class GraphTraversal { |
| 27 | +public final class PredecessorConstrainedDfs { |
28 | 28 |
|
29 |
| - private GraphTraversal() { |
| 29 | + private PredecessorConstrainedDfs() { |
30 | 30 | // utility class
|
31 | 31 | }
|
32 | 32 |
|
33 | 33 | /** An event emitted by the traversal: either a VISIT with an order, or a SKIP with a note. */
|
34 |
| - public static final class TraversalEvent<T> { |
35 |
| - private final T node; |
36 |
| - private final Integer order; // non-null for visit, null for skip |
37 |
| - private final String note; // non-null for skip, null for visit |
38 |
| - |
39 |
| - private TraversalEvent(T node, Integer order, String note) { |
40 |
| - this.node = node; |
41 |
| - this.order = order; |
42 |
| - this.note = note; |
| 34 | + public record TraversalEvent<T>( |
| 35 | + T node, |
| 36 | + Integer order, // non-null for visit, null for skip |
| 37 | + String note // non-null for skip, null for visit |
| 38 | + ) { |
| 39 | + public TraversalEvent { |
| 40 | + Objects.requireNonNull(node); |
| 41 | + // order and note can be null based on event type |
43 | 42 | }
|
44 | 43 |
|
45 | 44 | /** A visit event with an increasing order (0,1,2,...) */
|
46 | 45 | public static <T> TraversalEvent<T> visit(T node, int order) {
|
47 |
| - return new TraversalEvent<>(Objects.requireNonNull(node), order, null); |
| 46 | + return new TraversalEvent<>(node, order, null); |
48 | 47 | }
|
49 | 48 |
|
50 | 49 | /** A skip event with an explanatory note (e.g., not all parents visited yet). */
|
51 | 50 | public static <T> TraversalEvent<T> skip(T node, String note) {
|
52 |
| - return new TraversalEvent<>(Objects.requireNonNull(node), null, Objects.requireNonNull(note)); |
| 51 | + return new TraversalEvent<>(node, null, Objects.requireNonNull(note)); |
53 | 52 | }
|
54 | 53 |
|
55 | 54 | public boolean isVisit() {
|
56 | 55 | return order != null;
|
57 | 56 | }
|
| 57 | + |
58 | 58 | public boolean isSkip() {
|
59 | 59 | return order == null;
|
60 | 60 | }
|
61 |
| - public T node() { |
62 |
| - return node; |
63 |
| - } |
64 |
| - public Integer order() { |
65 |
| - return order; |
66 |
| - } |
67 |
| - public String note() { |
68 |
| - return note; |
69 |
| - } |
70 | 61 |
|
71 | 62 | @Override
|
72 | 63 | public String toString() {
|
73 |
| - return isVisit() ? "VISIT(" + node + ", order=" + order + ")" : "SKIP(" + node + ", " + note + ")"; |
| 64 | + return isVisit() |
| 65 | + ? "VISIT(" + node + ", order=" + order + ")" |
| 66 | + : "SKIP(" + node + ", " + note + ")"; |
74 | 67 | }
|
75 | 68 | }
|
76 | 69 |
|
@@ -118,56 +111,56 @@ public static <T> List<TraversalEvent<T>> dfsRecursiveOrder(Map<T, List<T>> succ
|
118 | 111 | return Collections.unmodifiableList(events);
|
119 | 112 | }
|
120 | 113 |
|
121 |
| - private static <T> void dfs(T u, Map<T, List<T>> succ, Map<T, List<T>> pred, Set<T> visited, int[] order, List<TraversalEvent<T>> out) { |
| 114 | + private static <T> void dfs(T currentNode, Map<T, List<T>> successors, Map<T, List<T>> predecessors, Set<T> visited, int[] order, List<TraversalEvent<T>> result) { |
122 | 115 |
|
123 |
| - if (!visited.add(u)) { |
| 116 | + if (!visited.add(currentNode)) { |
124 | 117 | return; // already visited
|
125 | 118 | }
|
126 |
| - out.add(TraversalEvent.visit(u, order[0]++)); // record visit and increment |
| 119 | + result.add(TraversalEvent.visit(currentNode, order[0]++)); // record visit and increment |
127 | 120 |
|
128 |
| - for (T v : succ.getOrDefault(u, List.of())) { |
129 |
| - if (visited.contains(v)) { |
| 121 | + for (T childNode : successors.getOrDefault(currentNode, List.of())) { |
| 122 | + if (visited.contains(childNode)) { |
130 | 123 | continue;
|
131 | 124 | }
|
132 |
| - if (allParentsVisited(v, visited, pred)) { |
133 |
| - dfs(v, succ, pred, visited, order, out); |
| 125 | + if (allParentsVisited(childNode, visited, predecessors)) { |
| 126 | + dfs(childNode, successors, predecessors, visited, order, result); |
134 | 127 | } else {
|
135 |
| - out.add(TraversalEvent.skip(v, "⛔ Skipping " + v + ": not all parents are visited yet.")); |
| 128 | + result.add(TraversalEvent.skip(childNode, "⛔ Skipping " + childNode + ": not all parents are visited yet.")); |
136 | 129 | // do not mark visited; it may be visited later from another parent
|
137 | 130 | }
|
138 | 131 | }
|
139 | 132 | }
|
140 | 133 |
|
141 |
| - private static <T> boolean allParentsVisited(T node, Set<T> visited, Map<T, List<T>> pred) { |
142 |
| - for (T p : pred.getOrDefault(node, List.of())) { |
143 |
| - if (!visited.contains(p)) { |
| 134 | + private static <T> boolean allParentsVisited(T node, Set<T> visited, Map<T, List<T>> predecessors) { |
| 135 | + for (T parent : predecessors.getOrDefault(node, List.of())) { |
| 136 | + if (!visited.contains(parent)) { |
144 | 137 | return false;
|
145 | 138 | }
|
146 | 139 | }
|
147 | 140 | return true;
|
148 | 141 | }
|
149 | 142 |
|
150 |
| - private static <T> boolean appearsAnywhere(Map<T, List<T>> succ, T node) { |
151 |
| - if (succ.containsKey(node)) { |
| 143 | + private static <T> boolean appearsAnywhere(Map<T, List<T>> successors, T node) { |
| 144 | + if (successors.containsKey(node)) { |
152 | 145 | return true;
|
153 | 146 | }
|
154 |
| - for (List<T> nbrs : succ.values()) { |
155 |
| - if (nbrs != null && nbrs.contains(node)) { |
| 147 | + for (List<T> neighbours : successors.values()) { |
| 148 | + if (neighbours != null && neighbours.contains(node)) { |
156 | 149 | return true;
|
157 | 150 | }
|
158 | 151 | }
|
159 | 152 | return false;
|
160 | 153 | }
|
161 | 154 |
|
162 |
| - private static <T> Map<T, List<T>> derivePredecessors(Map<T, List<T>> succ) { |
163 |
| - Map<T, List<T>> pred = new HashMap<>(); |
| 155 | + private static <T> Map<T, List<T>> derivePredecessors(Map<T, List<T>> successors) { |
| 156 | + Map<T, List<T>> predecessors = new HashMap<>(); |
164 | 157 | // ensure keys exist for all nodes appearing anywhere
|
165 |
| - for (Map.Entry<T, List<T>> e : succ.entrySet()) { |
166 |
| - pred.computeIfAbsent(e.getKey(), k -> new ArrayList<>()); |
167 |
| - for (T v : e.getValue()) { |
168 |
| - pred.computeIfAbsent(v, k -> new ArrayList<>()).add(e.getKey()); |
| 158 | + for (Map.Entry<T, List<T>> entry : successors.entrySet()) { |
| 159 | + predecessors.computeIfAbsent(entry.getKey(), key -> new ArrayList<>()); |
| 160 | + for (T childNode : entry.getValue()) { |
| 161 | + predecessors.computeIfAbsent(childNode, key -> new ArrayList<>()).add(entry.getKey()); |
169 | 162 | }
|
170 | 163 | }
|
171 |
| - return pred; |
| 164 | + return predecessors; |
172 | 165 | }
|
173 | 166 | }
|
0 commit comments