leetcode 2161.根据给定数字划分数组

1.题目要求

给你一个下标从 0 开始的整数数组 nums 和一个整数 pivot 。请你将 nums 重新排列，使得以下条件均成立：

所有小于 pivot 的元素都出现在所有大于 pivot 的元素 之前 。
所有等于 pivot 的元素都出现在小于和大于 pivot 的元素 中间 。
小于 pivot 的元素之间和大于 pivot 的元素之间的 相对顺序 不发生改变。
更正式的，考虑每一对 pi，pj ，pi 是初始时位置 i 元素的新位置，pj 是初始时位置 j 元素的新位置。对于小于 pivot 的元素，如果 i < j 且 nums[i] < pivot 和 nums[j] < pivot 都成立，那么 pi < pj 也成立。类似的，对于大于 pivot 的元素，如果 i < j 且 nums[i] > pivot 和 nums[j] > pivot 都成立，那么 pi < pj 。
请你返回重新排列 nums 数组后的结果数组。

 

示例 1：

输入：nums = [9,12,5,10,14,3,10], pivot = 10
输出：[9,5,3,10,10,12,14]
解释：
元素 9 ，5 和 3 小于 pivot ，所以它们在数组的最左边。
元素 12 和 14 大于 pivot ，所以它们在数组的最右边。
小于 pivot 的元素的相对位置和大于 pivot 的元素的相对位置分别为 [9, 5, 3] 和 [12, 14] ，它们在结果数组中的相对顺序需要保留。
示例 2：

输入：nums = [-3,4,3,2], pivot = 2
输出：[-3,2,4,3]
解释：
元素 -3 小于 pivot ，所以在数组的最左边。
元素 4 和 3 大于 pivot ，所以它们在数组的最右边。
小于 pivot 的元素的相对位置和大于 pivot 的元素的相对位置分别为 [-3] 和 [4, 3] ，它们在结果数组中的相对顺序需要保留。
 

题目代码:

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* pivotArray(int* nums, int numsSize, int pivot, int* returnSize) {
    int* less_number = (int*)malloc(sizeof(int) * numsSize);//小于pivot的数组
    int j_1 = 0;
    int* equal_number = (int*)malloc(sizeof(int) * numsSize);//等于pivot的数组
    int j_2 = 0;
    int* more_number = (int*)malloc(sizeof(int) * numsSize);//大于pivot的数组
    int j_3 = 0;
    //遍历nums数组,把nums中的元素按条件分给上面三个数组
    for(int i = 0;i < numsSize;i++){
        if(nums[i] < pivot){
            less_number[j_1] = nums[i];
            j_1++;
        }else if(nums[i] == pivot){
            equal_number[j_2] = nums[i];
            j_2++;
        }else{
            more_number[j_3] = nums[i];
            j_3++;
        }
    }
    int* number = (int*)malloc(sizeof(int) * numsSize);//重新排列后的数组
    int i = 0;
    int f = 0;
    //依次按要求把三个数组中的元素放入number数组中
    while(i < j_1){
        number[f] = less_number[i];
        f++;
        i++;
    }
    i = 0;
    while(i < j_2){
        number[f] = equal_number[i];
        f++;
        i++;
    }
    i = 0;
    while(i < j_3){
        number[f] = more_number[i];
        f++;
        i++;
    }
    *returnSize = numsSize;
    return number;
}

做题步骤都在代码里了，大家如果觉得好的话，不妨给个免费的赞吧，谢谢了^ _ ^