题解 | #牛的品种排序III# java

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param cows int整型一维数组
     * @param k int整型
     * @return int整型一维数组
     */
    public int[] sortCowsIII (int[] cows, int k) {
        // write code here
        int[] cnt = new int[k];
        for (int x : cows) {
            cnt[x] += 1;
        }
        int j = 0;
        for (int i = 0; i < k; i++) {
            while (cnt[i] > 0) {
                cows[j++] = i;
                cnt[i]--;
            }
        }
        return cows;
    }
}

Java代码

该题考察的知识点是计数排序。

代码中的sortCowsIII方法接受一个int[]类型的参数cows和一个整型参数k，并返回一个int[]类型的结果。

创建一个大小为k的新数组cnt，用于存储每个元素出现的次数。然后通过一次循环遍历输入的cows数组，并根据每个元素的值，将对应位置的计数加1。进行一次循环遍历cnt，通过内层循环将计数大于0的元素按照顺序放入cows数组的相应位置。内层循环中，通过不断递增j，从数组的起始位置开始填充元素。返回cows作为排序后的结果。