前言:
作者:神的孩子在歌唱
大家好,我叫智
75. 颜色分类
难度中等1028
给定一个包含红色、白色和蓝色,一共 n
个元素的数组,**原地**对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0
、 1
和 2
分别表示红色、白色和蓝色。
示例 1:
输入:nums = [2,0,2,1,1,0]
输出:[0,0,1,1,2,2]
示例 2:
输入:nums = [2,0,1]
输出:[0,1,2]
示例 3:
输入:nums = [0]
输出:[0]
示例 4:
输入:nums = [1]
输出:[1]
提示:
n == nums.length
1 <= n <= 300
nums[i]
为0
、1
或2
进阶:
- 你可以不使用代码库中的排序函数来解决这道题吗?
- 你能想出一个仅使用常数空间的一趟扫描算法吗?
package 排序;
import java.util.Arrays;
/*
* https://leetcode-cn.com/problems/sort-colors/
* 定义三个指针,red,white,blue,如果遇到white往前移动,
* 如果遇到0就和red交换,red++.如果遇到2就和blue交换,遇到1就继续white++
*/
public class _75_颜色分类 {
public void sortColors(int[] nums) {
int red=0,white=0,blue=nums.length-1;
// while循环,直到white超过blue
while(white<=blue) {
// 如果red为零就不用交换,直接++,并且将white放到red一起
if(nums[red]==0){
red++;
if(white<red){
white=red;
}
continue;
}
// 如果blue为2就不用交换,直接++
if(nums[blue]==2){
blue--;
continue;
}
// 如果white位置是2,那么就和blue位置交换
if(nums[white]==2){
sort(nums, white, blue);
blue--;
}
// 如果white位置是0,那么就和red位置交换
else if (nums[white]==0) {
sort(nums, white, red);
red++;
}
// 遇到1就++
else {
white++;
}
}
}
public void sort(int[] nums,int white,int num) {
int temp=nums[white];
nums[white]=nums[num];
nums[num]=temp;
}
}
本人csdn博客:https://blog.csdn.net/weixin_46654114
转载说明:跟我说明,务必注明来源,附带本人博客连接。