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]

解题思路

利用单指针，指向第一位。使用for循环，遍历每一位，如果有数字为0，则将它和指针所在位交换，指针加加；再使用for循环，判断为1的值，做相同操作。

代码

class Solution {
    public void sortColors(int[] nums) {
        int pre = 0;
        for(int i = 0;i < nums.length ; i++) {
            if(nums[i] == 0) {
                int temp = nums[i];
                nums[i] = nums[pre];
                nums[pre] = temp;
                pre++;
            }
        }

        for(int j = pre;j < nums.length;j++) {
            if(nums[j] == 1) {
                int temp = nums[j];
                nums[j] = nums[pre];
                nums[pre] = temp;
                pre++;
            }
        }
    }
}