以特定值C++拆分数组

#include <iostream>
#include <vector>

std::vector<int> splitArray(std::vector<int>& nums, int target) {
    int left = 0;
    int right = nums.size() - 1;

    while (left <= right) {
        while (left <= right && nums[left] < target) {
            left++;
        }
        while (left <= right && nums[right] >= target) {
            right--;
        }
        if (left <= right) {
            std::swap(nums[left], nums[right]);
            left++;
            right--;
        }
    }

    return nums;
}

int main() {
    std::vector<int> nums = {5, 2, 7, 3, 9, 1, 6};
    int target = 5;

    std::vector<int> result = splitArray(nums, target);

    std::cout << "Split Array: ";
    for (int num : result) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}