75颜色分类(荷兰国旗问题,三指针)

1、题目描述

给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。

进阶:

一个直观的解决方案是使用计数排序的两趟扫描算法。
首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗?

2、示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

3、题解

基本思想:荷兰国旗问题,三指针,一次遍历,如果是0,则移动到表头,如果是2,则移动到表尾,不用考虑1。这种思想和快速排序思想有点像,快速排序(Quick Sort)的基本思想是通过一趟排序将待排记录分割成独立的两部分,其中一部分记录的关键字均比另一部分记录的关键字小,则可分别对这两部分记录继续进行排序,以达到整个序列有序的目的。

  • left指向0的右边界,right指向2的左边界,也就是通过left和right将三色隔离红色[0,left)白色[left,right]蓝色(righ,n-1]
  • 当前元素为0,需要跟0的右边界left交换,0的右边界left++,cur++
  • 当前元素为1,cur一直位于left与right之间的1元素上,所以不需要交换,cur++
  • 当前元素为2,需要跟2的左边界right交换,2的左边界right--
#include<iostream>
#include<vector>
#include<algorithm>
#include<iterator>
using namespace std;
class Solution {
public:
    void sortColors(vector<int>& nums) {
        //基本思想:荷兰国旗问题,三指针,一次遍历,如果是0,则移动到表头,如果是2,则移动到表尾,不用考虑1
        //left指向0的右边界,right指向2的左边界,也就是通过left和right将三色隔离红色[0,left)白色[left,right]蓝色(righ,n-1]
        int left, right, cur;
        left = 0;
        right = nums.size() - 1;
        cur = 0;
        while (cur <= right)
        {
            //当前元素为0,需要跟0的右边界left交换,0的右边界left++,cur++
            if (nums[cur] == 0)
            {
                swap(nums[cur], nums[left]);
                left++;
                cur++;
            }
            //当前元素为1,cur一直位于left与right之间的1元素上,所以不需要交换,cur++
            else if (nums[cur] == 1)
            {
                cur++;
            }
            //当前元素为2,需要跟2的左边界right交换,2的左边界right--
            else
            {
                swap(nums[cur], nums[right]);
                right--;
            }
        }
        return;
    }
};
int main()
{
    Solution solute;
    vector<int> nums = { 2,0,2,1,1,0 };
    solute.sortColors(nums);
    copy(nums.begin(), nums.end(), ostream_iterator<int>(cout, " "));
    return 0;
}

 

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
荷兰国旗问题是一个经典的排序问题,它要求按照荷兰国旗颜色顺序(红、白、蓝)对一个包含红、白、蓝颜色的数组进行排序。 快速排序算法可以用来解决荷兰国旗问题。该算法基于分治的思想,通过多次划分和交换元素来达到排序的目的。以下是快速排序算法解决荷兰国旗问题的步骤: 1. 选择一个枢轴元素(一般是数组的最后一个元素)。 2. 初始化指针:low指向数组的起始位置,high指向数组的末尾位置,mid指向数组的起始位置。 3. 从头遍历数组,如果当前元素小于枢轴元素,则交换当前元素和mid指针指向的元素,并将mid指针后移一位。 4. 如果当前元素等于枢轴元素,则将high指针前移一位。 5. 如果当前元素大于枢轴元素,则交换当前元素和high指针指向的元素,并将high指针前移一位。 6. 重复步骤3到步骤5,直到low指针和high指针相遇为止。 7. 最后,将枢轴元素放在mid指针的位置上,这样数组就被分成了个部分:小于枢轴元素的部分、等于枢轴元素的部分和大于枢轴元素的部分。 8. 递归地对小于和大于枢轴元素的部分进行排序。 以下是一个示例荷兰国旗问题的快速排序的实现(使用Python语言): ```python def dutch_flag_sort(arr): def swap(arr, i, j): arr[i], arr[j] = arr[j], arr[i] def quicksort(arr, low, high): if low < high: pivot = arr[high] mid = low for i in range(low, high): if arr[i] < pivot: swap(arr, i, mid) mid += 1 swap(arr, mid, high) quicksort(arr, low, mid - 1) quicksort(arr, mid + 1, high) quicksort(arr, 0, len(arr) - 1) return arr # 示例用法 arr = [2, 0, 2, 1, 1, 0] sorted_arr = dutch_flag_sort(arr) print(sorted_arr) ``` 上述示例代码会输出 `[0, 0, 1, 1, 2, 2]`,即按照荷兰国旗顺序排序的结果。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值