Description
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.
Solution
n个这道题甚是简单,我的第一思路就是将所有的0换到前面,2换到后面,但是偷了懒,直接进行计数,而后对nums重新赋值。
代码实现如下:
class Solution {
public:
void sortColors(vector<int>& nums) {
int count[3] = { 0 };
for (int i = 0; i < nums.size(); i++)
count[nums[i]]++;
for (int i = 0; i < count[0]; i++)
nums[i] = 0;
for (int i = count[0]; i < count[0] + count[1]; i++)
nums[i] = 1;
for (int i = count[0] + count[1]; i < nums.size(); i++)
nums[i] = 2;
}
};
运行结果如下: