75.给定一个包含红色、白色和蓝色、共 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]
- 我们遍历时,把 0 往前放,把 2 往后放即可,这样就只需要两个指针记录放 0 和 2 的下标。
-
public void sortColors(int[] nums) { // left 记录放 0 的下标,right 记录放 2 的 int left=0,right=nums.length-1; // 遍历指针 int i=0; // 遍历到 i > right 时,right 后面的数都是 2,不用再调整位置了 while(i<=right){ // 遇到 0 就放最前面,然后 left+1,并且此时 i 对应的要么是 1 要么是 0 // 因为是 2 的话是最后一种情况的处理方式,早被放后面去了 // 如果 i 此时对应 0,那他肯定在正确位置了,i 直接 +1 // 如果 i 此时对应 1,那和第二种情况一样,不用管, i 直接 +1 if(nums[i]==0)swap(nums,left++,i++); else if(nums[i]==1)i++; // 如果从 right 换过来一个数,此时的 i 对应为 2 了 // 那我们肯定不能直接 i+1 把 2 留在错误的位置 // 所以 i 不 +1,下一轮接着判断 else swap(nums,right--,i); } } //交换数组中的两个数字 private void swap(int[] nums, int i, int j) { int tmp = nums[i]; nums[i] = nums[j]; nums[j] = tmp; }