Leetcode75. 颜色分类(C语言)

Leetcode75. 颜色分类(C语言)

算法-排序(荷兰国旗问题):算法与数据结构参考

题目:
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

思路:
给各个颜色计数,再按序赋值数组。

代码:

void sortColors(int* nums, int numsSize){
    int r=0,w=0,b=0;
    for(int i=0;i<numsSize;i++){
        if(nums[i]==0)  r++;
        else if(nums[i]==1)  w++;
        else  b++;
    }
    
    for(int j=0;j<r;j++)    nums[j]=0;
    for(int j=0;j<w;j++)    nums[j+r]=1;
    for(int j=0;j<b;j++)    nums[j+r+w]=2;
}

你可能感兴趣的:(数据结构&算法,算法,排序算法,c语言)