Update cat-and-mouse.cpp

lifei023 · Jan 17, 2021 · b2707f6 · b2707f6
1 parent ad21601
commit b2707f6
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/C++/cat-and-mouse.cpp b/C++/cat-and-mouse.cpp
@@ -33,13 +33,14 @@ class Solution {
         Lookup color;
         queue<tuple<int, int, int, int>> q;
         for(int i = 0; i < size(graph); ++i) {
+            if (i == HOLE) {
+                continue;
+            }
             for (const auto& t : {MOUSE, CAT}) {
                 color[make_tuple(HOLE, i, t)] = MOUSE;
                 q.emplace(HOLE, i, t, MOUSE);
-                if (i != HOLE) {
-                    color[make_tuple(i, i, t)] = CAT;
-                    q.emplace(i, i, t, CAT);
-                }
+                color[make_tuple(i, i, t)] = CAT;
+                q.emplace(i, i, t, CAT);
             }
         }
         while (!empty(q)) {