Sort Colors

最新推荐文章于 2021-04-15 12:30:39 发布

herogsep

最新推荐文章于 2021-04-15 12:30:39 发布

阅读量559

点赞数

分类专栏： LeetCode 文章标签： LeetCode

本文链接：https://blog.youkuaiyun.com/herogsep/article/details/9369781

版权

LeetCode 专栏收录该内容

2 篇文章

订阅专栏

这是一个编程题，要求对包含红色、白色和蓝色对象的数组进行排序，使得相同颜色的对象相邻，并按红色、白色、蓝色的顺序排列。题目禁止使用内置排序函数，解决策略是遍历数组，将0前移，2后移，确保遍历过程中i位置的元素不为0和2。

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

题目：

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.

思路：

从头到尾遍历数组，将0放到前面，将2放到后面，i所在的元素保证不为0也不为2，才i++。

class Solution {
public:
    void sortColors(int A[], int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        int p0=0, p2=n-1;
        int i = 0;
        for (i=0; i<=p2; i++)
        {
            if(A[i]==0)
            {
                A[i]=A[p0];
                A[p0++]=0;
            }
            else if(A[i]==2)
            {
                A[i]=A[p2];
                A[p2--]=2;
                i--;
            }
        }
        
    }
};