Sort Colors问题及解法

我们要爱学习

于 2017-08-01 18:28:53 发布

阅读量367

点赞数

分类专栏： LeetCode算法练习（Medium）文章标签： LeetCode Medium Two Pointers

本文链接：https://blog.youkuaiyun.com/u011809767/article/details/76549501

版权

LeetCode算法练习（Medium）专栏收录该内容

238 篇文章

订阅专栏

本文介绍了一种有效的三色排序算法，该算法可以将数组中表示红、白、蓝三种颜色的整数0、1、2进行排序，确保相同颜色的元素相邻且按红、白、蓝的顺序排列。通过两个指针分别从左右两端开始遍历，中间的元素作为待判断项，最终实现快速排序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述：

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.

问题分析：

对于2，我们可以一直放数组右边，对于0，我们可以一直放数组左边。

过程详见代码：

class Solution {
public:
    void sortColors(vector<int>& nums) {
       int right = nums.size() - 1;
		int left = 0;
		for (int i = 0; i <= right;i++)
		{
			while (nums[i] == 2 && i < right) swap(nums[i], nums[right--]);
			while (nums[i] == 0 && left < i) {swap(nums[i], nums[left++]); }
		} 
    }
};