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;
}