Update minimum-operations-to-remove-adjacent-ones-in-matrix.cpp

kamyu104 · web-flow · commit 6a00823c5cdf · 2022-01-02T13:05:10.000+08:00
diff --git a/C++/minimum-operations-to-remove-adjacent-ones-in-matrix.cpp b/C++/minimum-operations-to-remove-adjacent-ones-in-matrix.cpp
@@ -76,8 +76,8 @@ class Solution {
         if ((*lookup)[i][j]) {
             return;
         }
-        auto get_id = [&grid, &ids](int i, int j) {
-            const int x = size(grid[0]) * i + j, color = (i + j) % 2;
+        auto get_id = [&grid, &ids](int i, int j, int color) {
+            const int x = size(grid[0]) * i + j;
             if (!(*ids)[color].count(x)) {
                 (*ids)[color][x] = size((*ids)[color]);
             }
@@ -96,8 +96,8 @@ class Solution {
                     continue;
                 }
                 if (!color) {
-                    const int x = get_id(i, j);
-                    const int y = get_id(ni, nj);
+                    const int x = get_id(i, j, color);
+                    const int y = get_id(ni, nj, color ^ 1);
                     adj->resize(size((*ids)[0]));
                     (*adj)[x].emplace_back(y);
                 }

Original file line number	Diff line number	Diff line change
`@@ -76,8 +76,8 @@ class Solution {`
`76`	`76`	`if ((*lookup)[i][j]) {`
`77`	`77`	`return;`
`78`	`78`	`}`
`79`		`- auto get_id = [&grid, &ids](int i, int j) {`
`80`		`- const int x = size(grid[0]) * i + j, color = (i + j) % 2;`
	`79`	`+ auto get_id = [&grid, &ids](int i, int j, int color) {`
	`80`	`+ const int x = size(grid[0]) * i + j;`
`81`	`81`	`if (!(*ids)[color].count(x)) {`
`82`	`82`	`(ids)[color][x] = size((ids)[color]);`
`83`	`83`	`}`
`@@ -96,8 +96,8 @@ class Solution {`
`96`	`96`	`continue;`
`97`	`97`	`}`
`98`	`98`	`if (!color) {`
`99`		`- const int x = get_id(i, j);`
`100`		`- const int y = get_id(ni, nj);`
	`99`	`+ const int x = get_id(i, j, color);`
	`100`	`+ const int y = get_id(ni, nj, color ^ 1);`
`101`	`101`	`adj->resize(size((*ids)[0]));`
`102`	`102`	`(*adj)[x].emplace_back(y);`
`103`	`103`	`}`