Update wiggle-sort-ii.cpp

lifei023 · Jan 1, 2016 · 3180b1c · 3180b1c
1 parent bf549cd
commit 3180b1c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/C++/wiggle-sort-ii.cpp b/C++/wiggle-sort-ii.cpp
@@ -13,7 +13,8 @@ class Solution {
 
     void dutchFlagSort(vector<int>& nums, int val) {
         #define Nums(i) nums[(1 + 2 * (i)) % N]
-        for (int i = 0, j = 0, n = nums.size() - 1, N = nums.size() +  n % 2; j <= n;) {
+        const int N = nums.size() % 2 ? nums.size() : nums.size() + 1;
+        for (int i = 0, j = 0, n = nums.size() - 1; j <= n;) {
             if (Nums(j) > val) {
                 swap(Nums(i++), Nums(j++));
             } else if (Nums(j) < val) {