Update special-array-with-x-elements-greater-than-or-equal-x.cpp

kamyu104 · web-flow · commit cb36597a23ff · 2022-02-08T02:57:44.000+08:00
diff --git a/C++/special-array-with-x-elements-greater-than-or-equal-x.cpp b/C++/special-array-with-x-elements-greater-than-or-equal-x.cpp
@@ -50,15 +50,11 @@ class Solution2 {
             count[i] += count[i - 1];
         }
         for (int i = size(*nums) - 1; i >= 0; --i) {  // inplace but unstable sort
-            if ((*nums)[i] < 0) {  // processed
-                continue;
-            }
-            while (i != count[(*nums)[i]] - 1) {
+	        while ((*nums)[i] >= 0) {
                 --count[(*nums)[i]];
-                tie((*nums)[count[(*nums)[i]]], (*nums)[i]) = pair(~(*nums)[i], (*nums)[count[(*nums)[i]]]);
+                const int j = count[(*nums)[i]];
+                tie((*nums)[i], (*nums)[j]) = pair((*nums)[j], ~(*nums)[i]);
             }
-            --count[(*nums)[i]];
-            (*nums)[i] = ~(*nums)[i];
         }
         for (auto& num : *nums) {
             num = ~num;  // restore values

Original file line number	Diff line number	Diff line change
`@@ -50,15 +50,11 @@ class Solution2 {`
`50`	`50`	`count[i] += count[i - 1];`
`51`	`51`	`}`
`52`	`52`	`for (int i = size(*nums) - 1; i >= 0; --i) { // inplace but unstable sort`
`53`		`- if ((*nums)[i] < 0) { // processed`
`54`		`- continue;`
`55`		`- }`
`56`		`- while (i != count[(*nums)[i]] - 1) {`
	`53`	`+ while ((*nums)[i] >= 0) {`
`57`	`54`	`--count[(*nums)[i]];`
`58`		`- tie((nums)[count[(nums)[i]]], (nums)[i]) = pair(~(nums)[i], (nums)[count[(nums)[i]]]);`
	`55`	`+ const int j = count[(*nums)[i]];`
	`56`	`+ tie((nums)[i], (nums)[j]) = pair((nums)[j], ~(nums)[i]);`
`59`	`57`	`}`
`60`		`- --count[(*nums)[i]];`
`61`		`- (nums)[i] = ~(nums)[i];`
`62`	`58`	`}`
`63`	`59`	`for (auto& num : *nums) {`
`64`	`60`	`num = ~num; // restore values`