【刷题1】LeetCode 75. 颜色分类 java题解

题目

https://leetcode-cn.com/problems/sort-colors/

方法一：双指针

分析

设置两个指针p0(指向0要交换到的位置)，p1(指向1要交换到的位置)。
遍历数组，从i=0开始：
1.如果为0，和p0交换，p0右移一位，同时p1右移一位。
①如果p0<p1，说明已经将一些1连续地放在头部，此时一定会把一个1交换出去，导致答案错误。我们需要再交换nums[i]和nums[p1]。在进行了第一次交换之后nums[i]=1，我们需要将这个1放到头部的末端。
2.如果为1，和p1交换，p1右移一位。

代码

class Solution {
    public void sortColors(int[] nums) {
        int n=nums.length;
        int p0=0,p1=0;
        for(int i=0;i<n;i++){
            if(nums[i]==0){
                            //p0的位置是1，如果直接nums[i]换成p0,p0=0，那么就会少掉原来p0位置的1，需要补上一个1，1放在p1位置上，原来p1的数放在i
                if(p1>p0){
                    nums[i]=nums[p1];
                    nums[p1]=1;
                    nums[p0]=0;
                }
                else if(p1==p0){
                    nums[i]=nums[p0];
                    nums[p0]=0;
                }
                p0++;
                p1++;
            }
            else if(nums[i]==1){
                nums[i]=nums[p1];
                nums[p1]=1;
                p1++;
            }
        }
    }
}

复杂度

时间复杂度：O(n)，其中 n 是数组 nums 的长度。

空间复杂度：O(1)。

结果

方法二：单指针

分析

两次遍历

代码

class Solution {
    public void sortColors(int[] nums) {
        int n=nums.length;
        int head=0;
        for(int i=0;i<n;i++){
            if(nums[i]==0){
                nums[i]=nums[head];
                nums[head]=0;
                head++;
            }
        }
        int start=head;
        //head停留在一个不是0的数上，head应该是第一个1要放的位置
        for(int i=start;i<n;i++){
            if(nums[i]==1){
                nums[i]=nums[head];
                nums[head]=1;
                head++;
            }
        }
    }
}

复杂度

时间复杂度：O(n)，其中 n 是数组nums 的长度。

空间复杂度：O(1)。

结果