
很明显对于输入数组只需要采用排序就能得到答案,这里的整数限定了只能是0、1、2,那么采用计数排序可以很容易的解决
java
class Solution {
public void sortColors(int[] nums) {
int[] count = new int[3];
for (int num : nums) {
count[num]++;
}
int index = 0;
for (int i = 0; i < count.length; i++) {
while (count[i]-- != 0) {
nums[index++] = i;
}
}
}
}