题解 | #牛群的喂养顺序#

牛群的喂养顺序

https://www.nowcoder.com/practice/ce8860b6a8c74dfd8ccd15998970e447

知识点:栈,哈希表

已知有些元素需要有先后顺序,我们使用哈希表来存储序列关系,key为先放置的元素,value为后放置的元素,首先我们可以先排列没有前置条件的元素,然后对于每一个元素,判断该元素是否为前置元素,若是,则添加后置元素,重复以上步骤,直至没有后置元素可排列。具体来说,我们可以使用栈来实现,每入栈一个元素,判断栈顶元素是否为前置元素,然后将后置元素入栈,这样就可以只判断栈顶元素是否是前置元素即可。

Java题解如下

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param numCows int整型 
     * @param feedOrders int整型二维数组 
     * @return bool布尔型
     */
    public boolean canFeedAllCows (int numCows, int[][] feedOrders) {
        // write code here
        Map<Integer, Integer> map = new HashMap<>();
        boolean[] flag = new boolean[numCows];
        for(int[] order: feedOrders) {
            flag[order[0]] = true;
            map.put(order[1], order[0]);
        }
        Deque<Integer> stack = new ArrayDeque<>();
        for(int i = 0; i < numCows; i++) {
            if(!flag[i]) {
                stack.push(i);
            }
            while(map.containsKey(stack.peek())) {
                stack.push(map.get(stack.peek()));
            }
        }
        return stack.size() == numCows;
    }
}

全部评论

相关推荐

06-02 15:17
门头沟学院 Java
心爱的idea:怎么会呢 应该是打招呼有问题 问就说实习6个月全国可飞随时到岗
点赞 评论 收藏
分享
05-22 09:23
门头沟学院 Java
点赞 评论 收藏
分享
评论
点赞
1
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务