leetcode 75. Sort Colors

Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue.

Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

Note:
You are not suppose to use the library’s sort function for this problem.

排序O(Nlog(N))这道题是不行的。。
0 0 2 1 1 0 2 0 1 0 2
想了一下不知道是不是可以这样做:
左边都得是0 右边都得是2 1的位置是根据02个数确定的
一个左指针 遇到0就右走 遇到12的停住
一个右指针 遇到2就左走 遇到01就停住
那么此时可能是:
这里写图片描述
这样4种情况:
前3种比较好处理:

    1.两边调换之后 左指针走一步
    2.两边调换之后 左右指针都走一步
    3.两边调换之后 右指针走一步
    4.这时候要考虑 中间是不是都是1
        如果从左到右遍历一遍都是1 那么就结束
        如果发现了不是1的数(0或2)那么:
            发现的是0:和左指针对换 左指针走一步
            发现的是1:和右指针对换 右指针走一步
public void sortColors(int[] nums) {
        for(int i = 0,j = nums.length-1;i<j;){
            if(nums[i] == 0) i++;
            if(nums[j] == 2) j--;
            if(i >= j) break; 
            if(nums[i] == 2 && nums[j] == 0){
                swap(nums, i, j);
                i++;
                j--;
            }else if(nums[i] == 1 && nums[j] == 0){
                swap(nums, i, j);
                i++;
            }else if(nums[i] == 2 && nums[j] == 1){
                swap(nums, i, j);
                j--;
            }else{
                int k = i+1;
                for(;k < j;k++){
                    if(nums[k] == 0){
                        swap(nums, i, k);
                        i++;
                        break;
                    }
                    else if(nums[k] == 2){
                        swap(nums, k, j);
                        j--;
                        break;
                    }
                }
                if(k == j)
                    break;
            }
        }
    }

    public void swap(int[] nums,int i,int j) {
        int temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值