075 Sort Colors 分类颜色

给定一个包含红色、白色和蓝色,且含有 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

posted on 2018-04-03 17:14  lina2014  阅读(207)  评论(0编辑  收藏  举报

导航