leetcode：Sort Colors

标签：快速排序   索引   遍历   leetcode   

一、     题目

      给一个数组包含n个物体，有蓝色、红色和白色三种颜色，把他们分类并按照红、白、蓝的顺序排列，我们用0、1、2来表示红白蓝的颜色

注解：很容易想到遍历两遍数组得到三个数的数目，再覆盖，但是请只遍历一遍数组来解决。

二、     分析

      很简单，题目的意思其实就是让对一个数组排序，数组中的元素只有0、1、2，并且要求只能遍历一遍数组，常数空间复杂度。此时我们可以想到借助于快速排序的partition思想，以1为中间枢纽对数组进行划分，使0在数组的左边，2在数组的右边，1在数组的中间。   

class Solution {
public:
    void sortColors(int A[], int n) {
        int end = -1;
		int start = n;
		int i=0;
		//end是放0那部分的尾部索引，start是放2那部分的首部索引
        //碰到0放到end+1处，碰到2放到start-1处，碰到1指针直接后移
		while(i<start){
		  if(A[i] == 0 && i!=++end)
			 swap(A[end],A[i]);
		  else if(A[i] == 2 && i!= --start) 
			 swap(A[start],A[i]);
		  else
			 i++;
		}
    }
};

leetcode：Sort Colors

标签：快速排序   索引   遍历   leetcode   

原文地址：http://blog.csdn.net/zzucsliang/article/details/41397505