当前位置:网站首页>2161. 根据给定数字划分数组

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

2022-06-11 09:06:00 Mr Gao

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

给你一个下标从 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] ,它们在结果数组中的相对顺序需要保留。

这题,我们可以设置一个2*numsize的数组进行辅助,可以在不改变位置的情况下很好的解决问题
解题代码如下:

/** * Note: The returned array must be malloced, assume caller calls free(). */
int* pivotArray(int* nums, int numsSize, int pivot, int* returnSize){
    
    int *numst=(int *)malloc(sizeof(int)*numsSize*2);
    int epo=numsSize;
    int max=numsSize*2-1;
    int i=0;
    for(i;i<numsSize*2;i++){
    
        if(i<numsSize){
    
              numst[i]=nums[i];
        }
        else{
    
            numst[i]=-59789;
        }
    }
    for(i=numsSize-1;i>=0;i--){
    
        if(nums[i]>pivot){
    
            numst[max--]=nums[i];
            numst[i]=-59789;

        }
        if(nums[i]==pivot){
    
             numst[epo++]=nums[i];
             numst[i]=-59789;
        }
    }
    int po=0;
    for(i=0;i<numsSize*2;i++){
    
        if(numst[i]!=-59789){
    
    numst[po++]=numst[i];

        }
    }
    *returnSize=numsSize;
    return numst;
}
原网站

版权声明
本文为[Mr Gao]所创,转载请带上原文链接,感谢
https://blog.csdn.net/weixin_43327597/article/details/125207757