leetcode_sort_colors

发表于

难度：Medium

不使用迭代器了的做法。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class Solution {
public:
    void sortColors(vector<int>& nums) {
        int cur = 0;
        for(int i = 0; i < nums.size(); i++)
        {
            if(nums[i] == 0)
            {
                std::swap(nums[cur++],nums[i]);
            }
        }
        for(int i = cur; i < nums.size(); i++)
        {
            if(nums[i] == 1)
            {
                std::swap(nums[cur++],nums[i]);
            }
        }
    }
};

代码结果：运行时间3ms，超过12.96%。

解题思路：有两种解法。第一种：使用两个迭代器，第一个用来标识已经排好顺序的最后一位，第二个用来遍历，由于有0、1、2，需要遍历两次。第二种，计算0、1、2第一次出现的位置，出现的次数，然后遍历赋值。
代码如下。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution {
public:
    void sortColors(vector<int>& nums) {
        auto flag_it = nums.begin();
        int flag_value = 0;
        int two_loop = 2;
        while(two_loop-- && flag_it != nums.end())
        {
            auto start_it = flag_it;
            for(; start_it != nums.end(); start_it++)
            {
                if(*start_it == flag_value)
                {
                    swap(*(flag_it++),*start_it);
                }                
            }
            if(flag_it == nums.end())
                break;
            flag_value++;
        }
    }
};

代码结果：运行时间3ms，超过13.43%。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
class Solution {
public:
    void sortColors(vector<int>& nums) {
        auto zero_it = nums.end();
        auto one_it = nums.end();
        auto two_it = nums.end();
        int zero_number = 0, one_number = 0, two_number = 0;
        auto it = nums.begin();
        for(; it != nums.end(); it++)
        {
            if( *it == 0)
            {
                zero_number++;
                if(zero_it == nums.end())
                {
                    zero_it = it;
                }
            }
            else if( *it == 1)
            {
                one_number++;
                if(one_it == nums.end())
                {
                    one_it = it;
                }
            }
            else if( *it == 2)
            {
                two_number++;
                if(two_it == nums.end())
                {
                    two_it = it;
                }
            }
        }
        for(it = nums.begin(); it != nums.end(); it++)
        {
            if(zero_number>0)
            {
                *it = 0;
                zero_number--;
            }
            else
            {
                if(one_number>0)
                {
                    *it = 1;
                    one_number--;
                }
                else{
                    if(two_number>0)
                    {
                        *it = 2;
                        two_number--;
                    }
                }
            }
        }
    }
};

代码结果：运行时间3ms，超过13.43%。