题解 | #牛群的喂养顺序#
牛群的喂养顺序
https://www.nowcoder.com/practice/ce8860b6a8c74dfd8ccd15998970e447
import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param numCows int整型 * @param feedOrders int整型二维数组 * @return bool布尔型 */ public boolean canFeedAllCows(int numCows, int[][] feedOrders) { int[] feed = new int[numCows]; // 数组用于记录每头牛的饲料订单数量 int visited = 0; // 记录已访问并可以喂养的牛的数量 Queue<Integer> queue = new ArrayDeque<>(); // 队列用于存储可以喂养的牛的索引 List<List<Integer>> lists = new ArrayList<>(); // 列表用于存储每头牛的饲料订单 // 为每头牛初始化一个空的ArrayList for (int i = 0; i < numCows; i++) { lists.add(new ArrayList<>()); } // 填充feed数组和lists列表 for (int[] feedOrder : feedOrders) { feed[feedOrder[0]]++; // 增加该牛的饲料数量 lists.get(feedOrder[1]).add( feedOrder[0]); // 将该牛添加到对应索引的饲料订单列表中 } // 找到没有饲料订单的牛,并将其添加到队列中 for (int i = 0; i < numCows; i++) { if (feed[i] == 0) { queue.offer(i); visited++; } } // 处理队列中的牛,并更新依赖牛的饲料数量 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); // 如果依赖牛没有更多的饲料订单,将其添加到队列中 visited++; } } } return visited == numCows; // 如果所有的牛都被访问并可以喂养,则返回true,否则返回false } }
本题解题思路分析:
1.根据牛是否能被访问并喂养来判断是否为true或者false
2.先获得订单列表,增加牛饲料数量
3.将牛添加到对应索引的订单列表总
4.找到没有饲料的牛,添加到队列
5.处理队列中的牛,并更新依赖牛的饲料数量