一、题目
给定一个包含红色、白色和蓝色、共 n
个元素的数组 nums
,
原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
我们使用整数 0
、 1
和 2
分别表示红色、白色和蓝色。
必须在不使用库内置的 sort 函数的情况下解决这个问题。
示例 1:
输入:nums = [2,0,2,1,1,0]
输出:[0,0,1,1,2,2]
示例 2:
输入:nums = [2,0,1]
输出:[0,1,2]
提示:
n == nums.length
1 <= n <= 300
nums[i]
为0
、1
或2
二、思路解析
这道题我是用快排思想解决的,并且要用到三个指针,一个用来做中间变量,另外两个则是左右边界。
快排实际上就是,定义一个中间变量,然后让左右边界不断和这个变量比较。
如果排的是升序,则让小于中间变量的元素和左指针的元素交换,然后都进行右移操作;让大于中间变量的元素和右指针的元素交换,然后右指针进行左移操作,而中间变量指针不用移动,因为交换完的中间变量,也还没排好序,需要继续做中间变量。
而我们的目标是,经过处理之后,数组被分成如下四个部分:
编辑
这里还有一点细节需要注意的,就是左指针 left 和右指针 right 的初始值,要定义在数组之外。
初始化为 -1 是因为在开始时,还没有遇到任何 0,所以暂时将 “0区域” 的右边界设置在数组的最左边之外(可以想象成一个虚拟的位置),表明目前还没有 0 被放置在正确的位置。每当找到一个 0 (nums[i] == 0
),left
就会向右移动一位(通过 ++left
),并且当前的数字与 left
所指的位置交换,确保0被移到了数组的最前面。
而右指针也是同理。
编辑
三、完整代码
class Solution {
public void sortColors(int[] nums) {
int i = 0;
int left = -1;
int right = nums.length;
while(i < right){
if(nums[i] == 0){
swap(nums , ++ left , i ++);
}else if(nums[i] == 1){
i ++;
}else{
swap(nums , -- right , i);
}
}
}
public void swap(int[] nums ,int i , int j){
int tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
}
以上就是本篇博客的全部内容啦,如有不足之处,还请各位指出,期待能和各位一起进步!