【LeetCode刷题之旅】75 Sort Colors【C++】

最新推荐文章于 2024-07-30 16:17:15 发布

Jeloys

最新推荐文章于 2024-07-30 16:17:15 发布

阅读量218

点赞数 1

CC 4.0 BY-SA版权

分类专栏： LeetCode刷题之旅

本文链接：https://blog.csdn.net/weixin_38371360/article/details/86531206

LeetCode刷题之旅专栏收录该内容

12 篇文章

订阅专栏

本文对比了两种在数组中对颜色进行排序的算法——暴力解法和三路快速排序法。详细介绍了每种方法的实现过程，包括使用计数数组记录元素频率的暴力解法，以及仅遍历一次数组的三路快速排序法。尽管后者理论上更优，但实际运行时间可能更长。

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

听老师讲课，把所述的两种方法记录下来。

第一种是暴力解法：

class Solution {
public:
	//时间复杂度：O（n）
	//空间复杂度：O（k），k为元素的取值范围
	 
    void sortColors(vector<int>& nums) {
    	
    	int count[3] = {0};		//存放0，1，2三个元素的频率的数组
    	for (int i = 0; i < nums.size(); i++){
    		assert(nums[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;
		
		 
        
    }
};

第二种是三路快排：

class Solution {
public:
	//时间复杂度：O（n）
	//空间复杂度：O（1）
	//只遍历了一遍 
    void sortColors(vector<int>& nums) {
    	
    	int zero = -1;					// nums[0...zero] = 0,设置为-1表示此区间此时无效，没有任何一个元素
		int two = nums.size();			// nums[two...n-1] = 2,设置为nums.size()而不是num.size()-1，是因为此时无效
		for ( int i = 0; i< two ; ){
			if ( nums[i] == 1)
				i++;
			else if (nums[i] == 2){
				two --;
				swap( nums[i] , nums[two]);	//这两句也可合并成为： swap( nums[i] , nums[--two])
			}
				
			else{						// nums[i] == 0
				assert(nums[i] == 0);	//捕捉其他数值的异常输入
				zero ++;
				swap( nums[zero] , nums[i]);	//这三句可写成 swap( nums[++zero] , nums[i++])
				i++;				
			}
		} 
  	}
};

很奇怪，明明只遍历了一遍数组，时间却比暴力解法还要多。。。