Update count-connected-components-in-lcm-graph.cpp

Author: kamyu104

C++/count-connected-components-in-lcm-graph.cpp

@@ -42,6 +42,7 @@ class Solution {
     int countComponents(vector<int>& nums, int threshold) {
         UnionFind uf(threshold);
         vector<int> lookup(threshold, -1);
+        int result = size(nums);
         for (const auto& x : nums) {
             if (x - 1 >= threshold) {
                 continue;
@@ -51,18 +52,14 @@ class Solution {
                     lookup[i - 1] = x - 1;
                     continue;
                 }
-                uf.union_set(lookup[i - 1], x - 1);
+                if (uf.union_set(lookup[i - 1], x - 1)) {
+                    --result;
+                }
                 if (i == x) {
                     break;
                 }
             }
         }
-        int result = 0;
-        for (const auto& x : nums) {
-            if (x - 1 >= threshold || uf.find_set(x - 1) == x - 1) {
-                ++result;
-            }
-        }
         return result;
     }
 };