【leetcode】75. 颜色分类

给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

必须在不使用库的sort函数的情况下解决这个问题。

解题思路

单指针思路

代码

class Solution {
    public void sortColors(int[] nums) {
        int pre=0;//用pre来记录指针移动的位置
        //第一次遍历nums数组，将所有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++;
			}
		}
        //此时pre指向最后一个0的后节点
        //第二次遍历nums数组，将所有1交换到0的右边
		for(int i=pre;i<nums.length;i++) {
			if(nums[i]==1) {
				int temp=nums[i];
				nums[i]=nums[pre];
				nums[pre]=temp;
				pre++;
			}
		}
    }
}

 运行结果