题解 | #牛的品种排序III# java
牛的品种排序III
https://www.nowcoder.com/practice/f6ab3d7e20f54860886848f0a6374987
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
作为排序后的结果。