lc/lc75.cpp at main · Rikul/lc · GitHub

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
#include <vector>

using namespace std;

class Solution {
public:
    void sortColors(vector<int>& nums) {
        /*
        for(int i = 0; i < nums.size(); i++) {
            for(int j = i; j < nums.size(); j++) {
                if (nums[i] > nums[j]) {
                    int temp = nums[i];
                    nums[i] = nums[j];
                    nums[j] = temp;
                }
            }
        }*/

        vector<int> count = {0,0,0};
        for(int i = 0; i < nums.size(); i++) {
            count[nums[i]]++;
        }

        for(int i = 0; i < nums.size(); i++) {
            if (count[0] > 0) {
                nums[i] = 0; count[0]--;
            } else if (count[1] > 0) {
                nums[i] = 1; count[1]--;
            } else if (count[2] > 0) {
                nums[i] = 2; count[2]--;
            }
        }

    }
};