LeetCode 75. Sort Colors

题目

Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue.

Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

Note:
You are not suppose to use the library’s sort function for this problem.

Follow up:
A rather straight forward solution is a two-pass algorithm using counting sort.
First, iterate the array counting number of 0’s, 1’s, and 2’s, then overwrite array with total number of 0’s, then 1’s and followed by 2’s.

Could you come up with an one-pass algorithm using only constant space?

题意

给定一个n个对象的数组,对象颜色为红色,白色或蓝色,对它们进行排序,使相同颜色的对象相邻,颜色的顺序为红色,白色和蓝色。

在这里,使用0,1和2的整数分别表示红色,白色和蓝色。

注意:
不可以使用库的排序功能来解决这个问题。

跟进:
一个相当直接的解决方案是使用计数排序的双程算法。
首先,迭代数组的0,1和2的数字,然后覆盖总数为0的数组,然后是1,后跟2。

能不能相处一个只使用恒定空间的单遍算法吗?

    即,给出一个由0,1,2组成的数组,对它们进行排序,要求不能使用内置排序函数。

    跟进提到计算0,1,2出现的个数,然后按照这个个数覆盖原来的数组,这样的时间复杂度为两遍循环。

    那么有没有只用一个遍循环排序的算法呢?

分析

zero = 0, second = len-1。 // zero记录从前往后第一个值为0的位置,second记录从后往前第一个值为2的位置

遍历一遍数组

如果num[i] == 2, 把nums[i] 和nums[second]进行交换,如果等于0,和nums[zero]交换

代码

void sortColors(vector<int>& nums) {
    int second=nums.size()-1, zero=0;
    for (int i=0; i<=second; i++) {
        while (nums[i]==2 && i<second) swap(nums[i], nums[second--]);
        while (nums[i]==0 && i>zero) swap(nums[i], nums[zero++]);
    }
}

86 / 86 test cases passed.
Runtime: 6 ms

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值