Update partition-array-according-to-given-pivot.cpp

Author: kamyu104

C++/partition-array-according-to-given-pivot.cpp

@@ -5,16 +5,8 @@
 class Solution {
 public:
     vector<int> pivotArray(vector<int>& nums, int pivot) {
-        int less = 0, greater = 0;
-        for (const auto& x : nums) {
-            if (x < pivot) {
-                ++less;
-            } else if (x > pivot) {
-                ++greater;
-            }
-        }
         vector<int> result(size(nums), pivot);
-        int left = 0, right = size(nums) - greater;
+        int left = 0, right = size(nums) - count_if(cbegin(nums), cend(nums), [&pivot](const auto& x) { return x > pivot; });
         for (const auto& x: nums) {
             if (x < pivot) {
                 result[left++] = x;