From 584de7892550ff728365c8bd731dcbda24e6b2d7 Mon Sep 17 00:00:00 2001
From: Tusar Singh Nathawat
 <127783644+Tusarsinghnathawat@users.noreply.github.com>
Date: Wed, 16 Apr 2025 00:30:22 +0530
Subject: [PATCH] Update kruskal-mst.cpp

---
 03-Greedy/kruskal-mst.cpp | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/03-Greedy/kruskal-mst.cpp b/03-Greedy/kruskal-mst.cpp
index eff7bad..30f9479 100644
--- a/03-Greedy/kruskal-mst.cpp
+++ b/03-Greedy/kruskal-mst.cpp
@@ -54,7 +54,7 @@ vector<Edge> KruskalsMST(vector<Edge>& edges, int V) {
   }
 
   int i = 0; // Index used to pick next edge
-  while (result.size() < V - 1) {
+  for (const Edge& edge : edges) {
     // Pick next edge from the sorted list
     Edge next_edge = edges[i++];
 
@@ -84,9 +84,13 @@ int main() {
   vector<Edge> mst = KruskalsMST(edges, V);
 
   cout << "Edges in the constructed MST: \n";
-  for (Edge edge : mst) {
+  for (const Edge& edge : mst) {
     cout << edge.src << " -- " << edge.dest << " == " << edge.weight << endl;
   }
+  if (mst.size() < V - 1) {
+    cout << "\nNote: The graph is disconnected. Forest generated instead of a single MST.\n";
+  }
+
 
   return 0;
 }