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