more golf

kth-competitive-programming · Oct 23, 2024 · 53a1582 · 53a1582
1 parent d81bf6e
commit 53a1582
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/content/graph/TopoSort.h b/content/graph/TopoSort.h
@@ -15,11 +15,7 @@ vi topoSort(const vector<vi>& gr) {
 	vi indeg(sz(gr)), q;
 	for (auto& li : gr) for (int x : li) indeg[x]++;
 	rep(i,0,sz(gr)) if (indeg[i] == 0) q.push_back(i);
-	int j = 0;
-	while (j < sz(q)) {
-		int i = q[j++];
-		for (int x : gr[i])
-			if (--indeg[x] == 0) q.push_back(x);
-	}
+	rep(j,0,sz(q)) for (int x : gr[q[j]])
+		if (--indeg[x] == 0) q.push_back(x);
 	return q;
 }