Java 题解 | #牛群的喂养顺序II#
牛群的喂养顺序II
https://www.nowcoder.com/practice/05abc133293a42358bbbb4016034728e
import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param numCows int整型 * @param feedOrders int整型二维数组 * @return int整型一维数组 */ public int[] findFeedOrderII (int numCows, int[][] feedOrders) { // write code here int[] feed = new int[numCows]; int visited = 0; List<Integer> res = new ArrayList<>(); Queue<Integer> queue = new LinkedList<>(); List<List<Integer>> lists = new ArrayList<>(); for (int i = 0; i < numCows; i++) { lists.add(new ArrayList<>()); } for (int[] feeds : feedOrders) { feed[feeds[0]]++; lists.get(feeds[1]).add(feeds[0]); } for (int i = 0; i < numCows; i++) { if (feed[i] == 0) { queue.offer(i); visited++; res.add(i); } } while (!queue.isEmpty()) { int index = queue.poll(); List<Integer> list = lists.get(index); for (int i : list) { feed[i]--; if (feed[i] == 0) { queue.offer(i); res.add(i); visited++; } } } if (visited == numCows) { int[] result = new int[res.size()]; for (int i = 0; i < res.size(); i++) { result[i] = res.get(i); } return result; } else { return new int[0]; } } }
该代码使用的编程语言是Java。
该题表达的知识点:
- 数组:使用
int[]
表示整数数组。 - 列表:使用
List
和ArrayList
来动态管理列表。 - 队列:使用
Queue
和LinkedList
来实现队列数据结构。 - 二维数组和二维列表:使用
int[][]
和List<List<Integer>>
表示二维整数数组和二维列表。 - 循环:使用
for
循环遍历数组和列表。 - 添加元素:使用
list.add(element)
将元素添加到列表中。 - 检查队列是否为空:使用
queue.isEmpty()
判断队列是否为空。 - 入队和出队操作:使用
queue.offer(element)
向队列中添加元素,使用queue.poll()
从队列中取出并删除第一个元素。 - 转换数组和列表:使用
.toArray()
方法将列表转换为数组。
代码的文字解释大纲如下:
- 定义一个函数
findFeedOrderII
,参数为牛的数量numCows
和喂养订单的二维数组feedOrders
。 - 创建一个整数数组
feed
,用于记录每头牛的喂养次数,默认都为0。 - 创建一个整数变量
visited
,用于记录已访问的牛的数量。 - 创建一个列表
res
,用于记录按照喂养顺序的牛的索引。 - 创建一个队列
queue
,用于存放待访问的牛的索引。 - 创建一个二维列表
lists
,用于记录每个饲料的制造者所喂养的牛的索引。 - 遍历
feedOrders
数组,统计每头牛的喂养次数,并将对应的牛的索引添加到lists
列表中。 - 遍历
feed
数组,找出初始时没有被喂养过的牛,将其索引加入队列、visited
自增并添加到res
列表中。 - 使用
while
循环,当队列不为空时进行以下操作:出队一个牛的索引index,并获取对应的饲料制造者列表list。遍历list中的每个制造者的索引i,更新对应牛的喂养次数feed[i]--。如果更新后的喂养次数为0,将该牛的索引入队、visited自增并添加到res列表中。 - 判断是否所有牛都被喂养过,如果是,则将
res
列表转换为数组返回;否则,返回空数组。