leetcode2161 根据给定数字划分数组

一、问题描述

二、解题思路

本题与荷兰国旗问题有一定的区别，如果用三指针来解决，相对顺序是有可能被打乱的，所以本题可以使用直白的三次遍历，即暴力，分别将<pivot的数，=pivot的数，>pivot的数加入ret向量。

三、代码实现

class Solution {
public:
    vector<int> pivotArray(vector<int>& nums, int pivot) {
        //3次遍历
        vector<int> ret;
        for(int i=0;i!=nums.size();i++)
            if(nums[i]<pivot) ret.push_back(nums[i]);
        for(int i=0;i!=nums.size();i++)
            if(nums[i]==pivot) ret.push_back(nums[i]);
        for(int i=0;i!=nums.size();i++)
            if(nums[i]>pivot) ret.push_back(nums[i]);

        return ret;

    }
};