75. Sort Colors

这个题的意思是,给一个包含0,1,2无序的数组,排序。

我的方法是,遍历两遍数组,记录0的个数,1的个数,2的个数,然后重写数组。其中定义一个idx,记录下现在写到哪一个了。

class Solution {
public:
    void sortColors(vector<int>& nums) {
        unordered_map<int, int> m;
        int idx = 0;
        for(int i = 0; i < nums.size(); ++i){
            m[nums[i]]++;
        }
        for(int i = 0; i < 3; ++i){
            for(int j = 0; j < m[i]; ++j){
                nums[idx++] = i;
            }
        }
    }
};

题目中还要让只遍历一次数组来求解,那么我需要用双指针来做,分别从原数组的首尾往中心移动。

- 定义red指针指向开头位置,blue指针指向末尾位置

- 从头开始遍历原数组,如果遇到0,则交换该值和red指针指向的值,并将red指针后移一位。若遇到2,则交换该值和blue指针指向的值,并将blue指针前移一位。若遇到1,则继续遍历。这样就把0都移到数组的左端,2都移到数组的右端了。

class Solution {
public:
    void sortColors(int A[], int n) {int red = 0, blue = n - 1;
        for (int i = 0; i <= blue; ++i) {
            if (A[i] == 0) {
                swap(A[i], A[red++]);
            } else if (A[i] == 2) {
                swap(A[i--], A[blue--]);
            } 
        }
    }
};



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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值