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

最新推荐文章于 2024-05-13 20:35:22 发布

shui_haiyang

最新推荐文章于 2024-05-13 20:35:22 发布

阅读量300

点赞数

分类专栏：算法学习记录文章标签： leetcode c++ 算法

本文链接：https://blog.csdn.net/shui_haiyang/article/details/131880514

版权

算法学习记录专栏收录该内容

14 篇文章

订阅专栏

该文章讨论了一种针对包含红、白、蓝三种颜色元素的数组排序问题，需按颜色顺序排列且不使用内置排序函数。这个问题类似于荷兰国旗问题，通过使用三个指针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;
}