力扣------根据给定数字划分数组

最新推荐文章于 2024-07-16 23:15:32 发布

qq_37760746

最新推荐文章于 2024-07-16 23:15:32 发布

阅读量71

点赞数

分类专栏： C/C++ 文章标签： leetcode 排序算法算法

本文链接：https://blog.csdn.net/qq_37760746/article/details/125141335

版权

C/C++ 专栏收录该内容

114 篇文章 0 订阅

订阅专栏

在这里插入图片描述

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* pivotArray(int* nums, int numsSize, int pivot, int* returnSize){
    *returnSize=numsSize;
    int lessthan[100000];
    int top1=0;
    int beequalto[100000];
    int top2=0;
    int greaterthan[100000];
    int top3=0;
    for(int a=0;a<numsSize;a++){
        if(nums[a]<pivot){
            lessthan[top1++]=nums[a];
        }
        else if(nums[a]==pivot){
            beequalto[top2++]=nums[a];
        }
        else{
            greaterthan[top3++]=nums[a];
        }
    }
    for(int a=0;a<top1;a++){
        nums[a]=lessthan[a];
    }
    for(int a=top1;a<top1+top2;a++){
        nums[a]=beequalto[a-top1];
    }
    for(int a=top1+top2;a<top1+top2+top3;a++){
        nums[a]=greaterthan[a-top1-top2];
    }
    return nums;
}