Leet Code Daily 2024-06-12 - Sort Colors
Code
class Solution {
public:
void sortColors(vector<int>& nums) {
int cnt[3] = {0};
for (const auto& x : nums) {
cnt[x]++;
}
for (auto& x : nums) {
for (int v = 0; v < 3; v++) {
if (cnt[v]-- > 0) {
x = v;
break;
}
}
}
}
};