|
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