[刷题] 75 Sort Colors

要求

• 给只有0 1 2三个元素的数组排序

思路

• 方法1：遍历数组，利用辅助数组保存三个元素的个数，再写入（遍历两遍）
  • 辅助数组有三个元素，对应0 1 2的个数
• 方法2：模拟三路快排，遍历一遍完成排序
  • 三个索引，zero和two用于首尾的扩充，i用于遍历
  • 遍历数组，每个元素只有三种可能
  • 0 2移到首尾，1不动

实现

方法1

void sortColors(vector<int>& nums){
    int count[3] = {0};
    for( int i = 0 ; i < nums.size() ; i ++ ){
        assert( num[i] >= 0 && nums[i] <= 2 );
        count[nums[i]] ++;
    }
            
    int index = 0;
    for( int i = 0 ; i < count[0] ; i ++ ){
        nums[index++] = 0;
    for( int i = 0 ; i < count[1] ; i ++ ){
        nums[index++] = 1;
    for( int i = 0 ; i < count[2] ; i ++ ){
        nums[index++] = 2;
    }
}

方法2

void sortColors1(vector<int>& nums){
    int zero = -1;
    int two = nums.size();
    for( int i = 0 ; i < two ; ){
        if( nums[i] == 1 )
            i ++;
        else if( nums[i] == 2 ){
            two --;
            swap( nums[i] , nums[two] );    
        }else{
            assert( nums[i] == 0 );
            zero ++;
            swap( nums[zero] , nums[i] );
            i ++;
        }
    }
}

相关

215 Kth Largest Element in an Array