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

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param cows int整型vector 
     * @return int整型vector
     */
    vector<int> sortCows(vector<int>& cows) {
        // write code here
        int n = cows.size();
        vector<int> c(n,0);
        for(int i=0;i<n;i++){
            c[cows[i]]++;
        }
        for(int i=1;i<3;i++){
            c[i] += c[i-1];
        }
        vector<int> r(n);
        for(int i=n-1;i>=0;i--){
            int index = c[cows[i]]-1;
            r[index] = cows[i];
            c[cows[i]]--;
        }
        for(int i=0;i<n;i++){
            cows[i] = r[i];
        }
        return cows;
    }
};