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列表转换为数组返回;否则,返回空数组。

