给定一个包含红色、白色和蓝色、共n个元素的数组nums,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排序。
使用整数0、1和2分布表示红色、白色和蓝色。
必须在不使用库内置sort函数的情况下解决这个问题。
示例1:
cs输入:nums = [2,0,2,1,1,0] 输出:[0,0,1,1,2,2]
示例2:
cs输入:nums = [2,0,1] 输出:[0,1,2]
提示:
n == nums.length
1 <= n <= 300
nums[i]
为0
、1
或2
- 进阶:你能想出一个仅使用常数空间的一趟扫描算法吗?
本题是经典的【荷兰国旗问题】,由计算机科学家 Edsger W. Dijkstra 首先提出。
解题思路:方法一:单指针
**两次遍历:**在第一次遍历中,将数组中所有的0交换到数组头部。
第二次遍历中,将数组中所有的1交换到头部的0之后。
csvoid swap(int *a,int *b) { int t = *a; *a = *b,*b = t; } void sortColors(int *nums,int numsSize) { int ptr = 0; for(int i=0;i<numsSize;++i) { if(nums[i]==0) { swap(&nums[i],&nums[ptr]); ++ptr; } } for(int i=ptr;i<numsSize;i++) { if(nums[i]==1) { swap(&nums[i],&nums[ptr]); ++ptr; } } }
时间复杂度:O(n),其中n是数组nums的长度。
空间复杂度:O(1)
方法二:双指针
使用两个指针分别来交换0和1
csvoid swap(int *a,int *b) { int t= *a; *a= *b,*b=t; } void sortColors(int *nums,int numsSize) { int p0 = 0,p1=0; for(int i=0;i<numsSize;i++) { if(nums[i]==1) { swap(&nums[i],&nums[p1]); p1++; }else if(nums[i]==0) { swap(&nums[i],&nums[p0]); if(p0<p1) swap(&nums[i],&nums[p1]); ++p0,++p1; } } }
时间复杂度:O(n),其中n是数组nums的长度
空间复杂度:O(1)
方法三:双指针
左指针P0来交换0
右指针P2来交换2
csvoid swap(int *a,int *b) { int t=*a; *a=*b,*b=t; } void sortColors(int *nums,int numsSize) { int p0=0,p2=numsSize-1; for(int i=0;i<p2;i++) { while(i<=p2 && nums[i]==2) { swap(&nums[i],&nums[p2]); --p2; } if(nums[i]==0) { swap(&nums[i],&nums[p0]); ++p0; } } }
时间复杂度:O(n),其中n是数组nums的长度
空间复杂度:O(1)
方法四:记录0 1 2的个数,对其进行赋值即可。
LeetCode 第75题:颜色分类
IC 见路不走2025-06-16 8:21
相关推荐
沐怡旸2 小时前
【算法】【链表】328.奇偶链表--通俗讲解掘金安东尼5 小时前
Amazon Lambda + API Gateway 实战,无服务器架构入门码流之上6 小时前
【一看就会一写就废 指间算法】设计电子表格 —— 哈希表、字符串处理快手技术8 小时前
快手提出端到端生成式搜索框架 OneSearch,让搜索“一步到位”!CoovallyAIHub1 天前
中科大DSAI Lab团队多篇论文入选ICCV 2025,推动三维视觉与泛化感知技术突破NAGNIP1 天前
Serverless 架构下的大模型框架落地实践moonlifesudo1 天前
半开区间和开区间的两个二分模版moonlifesudo1 天前
300:最长递增子序列CoovallyAIHub1 天前
港大&字节重磅发布DanceGRPO:突破视觉生成RLHF瓶颈,多项任务性能提升超180%!CoovallyAIHub1 天前
英伟达ViPE重磅发布!解决3D感知难题,SLAM+深度学习完美融合(附带数据集下载地址)