题解 | #牛的品种排序II#

知识点

暴力

思路

先遍历一遍数组，统计黑黄白出现的次数，然后直接更新到答案数组里即可。

代码

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param cows int整型vector 
     * @return int整型vector
     */
    vector<int> sortCows(vector<int>& cows) {
        // write code here
        vector<int>ans;
        int b=0,w=0,h=0;
        for(auto v:cows)
        {
            if(v==1)w++;
            else if(v==2)h++;
            else b++;
        }
        for(int i=1;i<=b;i++)ans.push_back(0);
        for(int j=1;j<=w;j++)ans.push_back(1);
        for(int j=1;j<=h;j++)ans.push_back(2);
        return ans;
    }
};