0924 codes updated.

liuyubobobo · liuyubobobo · commit c094f1904c15 · 2019-09-23T06:59:16.000-07:00
diff --git a/0924-Minimize-Malware-Spread/cpp-0924/main.cpp b/0924-Minimize-Malware-Spread/cpp-0924/main.cpp
@@ -32,8 +32,8 @@ class Solution {
         visited.resize(n, false);
 
         int best = 0, res = -1, ccid = 0;
-        for(int v = 0; v < n; v ++)
-            if(init_set.count(v) && !visited[v]){
+        for(int v: init_set)
+            if(!visited[v]){
                 cc_sz.push_back(0), cc_mal_num.push_back(0);
                 cc_sz[ccid] = dfs(graph, v, ccid);
                 if(cc_mal_num[ccid] == 1 && cc_sz[ccid] > best){
@@ -50,6 +50,7 @@ class Solution {
 
         visited[v] = true;
         if(init_set.count(v)) cc_mal_num[ccid] ++;
+
         int res = 1;
         for(int next = 0; next < n; next ++)
             if(g[v][next] && !visited[next])
diff --git a/0924-Minimize-Malware-Spread/cpp-0924/main2.cpp b/0924-Minimize-Malware-Spread/cpp-0924/main2.cpp
@@ -67,8 +67,7 @@ class Solution {
                 if(graph[i][j])
                     uf.unionElements(i, j);
 
-        set<int> init_set;
-        for(int e: initial) init_set.insert(e);
+        set<int> init_set(initial.begin(), initial.end());
 
         unordered_map<int, int> cc_sz, cc_mal_num;
         for(int e: init_set)
