C++ 国旗问题&颜色分类 LeetCode 75题

该文章讨论了一种针对包含红、白、蓝三种颜色元素的数组排序问题,需按颜色顺序排列且不使用内置排序函数。这个问题类似于荷兰国旗问题,通过使用三个指针small_p,big_p,move_p,实现了原地排序。算法中,small_p指向小于target的元素,big_p指向大于target的元素,move_p遍历数组。通过移动指针并交换元素,最终达到目标排序状态。

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

题目描述

给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。

我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

必须在不使用库内置的 sort 函数的情况下解决这个问题。

荷兰国旗问题与上述题目类似,可用同套代码解答

问题描述:
给定一个数组nums,和一个数target,请把小于等于target的数放在数组的左边,大于target的数放在数组的右边。要求额外空间复杂度O(1),时间复杂度 O(N)

思路

准备三个指针
第一个指针small_p 指向比target小的区域
第二个指针big_p 指向比target大的区域
第三个指针move_p 指向数组起始位置,用于遍历数组

数组左右各一个指针,分别代表小区域和大区域,通过move_p和target比较的情况,不停的让左右指针向中间移动

如果move_p < target small_p向后移动一位,然后交换small_p 和 move_p,再让move_p向后移动一位。
如果move_p = target 只需要把move_p 向后移动一位即可
如果move_p > target 把big_p向前移动一位,然后交换big_p 和 move_p

vector<int> france_flag(vector<int>& nums, int target)
{
	//小于target指针
	int small_p = -1;
	//大于target指针
	int big_p = nums.size();
	//遍历数组的指针
	int move_p = 0;

	while(move_p < big_p)
	{
		if(nums[move_p] < target)
		{
			small_p++;
			swap(nums, move_p, small_p);
			move_p++;
		}
		else if(nums[move_p] == target)
		{
			move_p++;
		}
		else
		{
			big_p--;
			swap(nums, move_p, big_p);
		}
	}

	return nums;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值