[LeetCode]Sort Colors

最新推荐文章于 2019-05-21 15:41:52 发布

原创最新推荐文章于 2019-05-21 15:41:52 发布 · 439 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

149 篇文章

订阅专栏

本文介绍了一种高效的三色排序算法，适用于将红色、白色和蓝色的对象按顺序排列。该算法利用三个指针来分别标记左边界、右边界和当前元素的位置，并通过交换元素的方式将相同颜色的对象聚集在一起。整个过程无需使用额外的排序库，时间复杂度为O(N)。

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.

LeetCode Source

分析：对0，1，2排序，用3个指针，一个指到最右边，一个指到最左边，一个指当前对象。如果是0往最左边移动，如果是2往最右边移动。移动后对应边界改变。O(N)复杂度。

class Solution {
public:
    void sortColors(int A[], int n) { //use three point , point right ,left,and curr
      int left = 0;
      int right = n-1;
      int curr = left;
      while(curr<=right){
          curr = (curr<=left)?left:curr;
          if(A[curr]==0){
              swap(A,left++,curr++);
          }
          else if(A[curr]==2){
              swap(A,right--,curr);
          }
          else{
              ++curr;
          }
      }
        
    }
    void swap(int A[],int a,int b){
        int temp = A[b];
        A[b] = A[a];
        A[a] = temp;
    }
};