Update minimum-swaps-to-group-all-1s-together-ii.cpp

kamyu104 · web-flow · commit d696285dd4a5 · 2022-01-09T23:03:36.000+08:00
diff --git a/C++/minimum-swaps-to-group-all-1s-together-ii.cpp b/C++/minimum-swaps-to-group-all-1s-together-ii.cpp
@@ -6,7 +6,7 @@ class Solution {
     int minSwaps(vector<int>& nums) {
         const int w = count(cbegin(nums), cend(nums), 1);
         int result = w, cnt = w;
-        for (int i = 0; i < size(nums) * 2; ++i) {
+        for (int i = 0; i < size(nums) + (w - 1); ++i) {
             if (i >= w) {
                 cnt += nums[(i - w) % size(nums)];
             }

Original file line number	Diff line number	Diff line change
`@@ -6,7 +6,7 @@ class Solution {`
`6`	`6`	`int minSwaps(vector<int>& nums) {`
`7`	`7`	`const int w = count(cbegin(nums), cend(nums), 1);`
`8`	`8`	`int result = w, cnt = w;`
`9`		`- for (int i = 0; i < size(nums) * 2; ++i) {`
	`9`	`+ for (int i = 0; i < size(nums) + (w - 1); ++i) {`
`10`	`10`	`if (i >= w) {`
`11`	`11`	`cnt += nums[(i - w) % size(nums)];`
`12`	`12`	`}`