leetcode 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]
提示:
-
n == nums.length
-
1 <= n <= 300
-
nums[i]
为0
、1
或2
进阶:
-
你可以不使用代码库中的排序函数来解决这道题吗?
-
你能想出一个仅使用常数空间的一趟扫描算法吗?
Related Topics
数组
双指针
排序
思路1:双指针
使用2个指针下标
-
第一遍遍历,把0放在最左边。
-
第二遍遍历,把1放在中间。
class Solution { public void sortColors(int[] nums) { int left = sortColors(nums,0,nums.length-1,0); sortColors(nums,left,nums.length-1,1); } public int sortColors(int[] nums,int left,int right,int target){ //把0放到左边 while(left <= right){ if(nums[left] == target){ left++; }else if(nums[right]!=target){ right--; }else{//nums[left]!=0 && nums[right]==0 进行交换 swap(nums,left,right); left++; right--; } } return left; } public void swap(int[] nums,int i,int j){ int temp = nums[i]; nums[i] = nums[j]; nums[j] = temp; } } 解答成功: 执行耗时:0 ms,击败了100.00% 的Java用户 内存消耗:39.6 MB,击败了33.73% 的Java用户
另外一种形式:如果颜色比较多,可以使用上述方法进行循环迭代。
class Solution { public void sortColors(int[] nums) { int left = 0; for(int i = 0 ; i < nums.length;i++){ if(nums[i]==0){ swap(nums,left,i); left++; } } for(int i = left;i<nums.length;i++){ if (nums[i] == 1){ swap(nums,left,i); left++; } } } public void swap(int[] nums,int i,int j){ int temp = nums[i]; nums[i] = nums[j]; nums[j] = temp; } }