给定一个包含红色、白色和蓝色,且含有 n 个元素的数组,对它们进行排序,使得相同颜色的元素相邻,颜色顺序为红色、白色、蓝色。
此题中,我们使用整数 0, 1 和 2 分别表示红色,白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
进阶:
一个相当直观的解决方案是使用计数排序的 two-pass 算法。
首先,迭代计算出0,1 和 2 元素的个数,然后重写当前数组。
你能想出一个仅使用恒定空间的 one-pass 算法吗?
详见:https://leetcode.com/problems/sort-colors/description/
Java实现:
方法一:计数排序思想
class Solution { public void sortColors(int[] nums) { int i=0; int j=0; int k=0; for(int p=0;p<nums.length;++p){ if(nums[p]==0){ ++i; }else if(nums[p]==1){ ++j; }else{ ++k; } } for(int p=0;p<nums.length;++p){ if(p<i){ nums[p]=0; }else if(p>=i&&p<(i+j)){ nums[p]=1; }else{ nums[p]=2; } } } }
方法二:
设置两个index,left记录第一个1的位置,left左边为0,right记录第一个非2的位置,right右边为2.
然后使用i从头到尾扫一遍,直到与right相遇。
i遇到0就换到左边去,遇到2就换到右边去,遇到1就跳过。
需要注意的是:由于left记录第一个1的位置,因此nums[left]与nums[i]交换后,nums[left]为0,nums[i]为1,因此i++;
而right记录第一个非2的位置,可能为0或1,因此nums[right]与nums[i]交换后,nums[right]为2,nums[i]为0或1,i不能前进,要后续判断。
由此该数组分为4段:[0,left)-->0; [left,i)-->1; [i,right]-->乱序; (right,n-1]-->2
class Solution { public void sortColors(int[] nums) { int left=0; int right=nums.length-1; int i=0; while(i<=right){ if(nums[i]==0){ swap(nums,i,left); ++left; ++i; }else if(nums[i]==1){ ++i; }else{ swap(nums,i,right); --right; } } } private void swap(int[] nums,int i,int j){ int tmp=nums[i]; nums[i]=nums[j]; nums[j]=tmp; } }
参考:https://www.cnblogs.com/ganganloveu/p/3703746.html