LintCode 618.搜索图中节点

给定一张 无向图,一个节点以及一个目标值,返回距离这个节点最近且值为目标值的节点,如果不能找到则返回 NULL。
在给出的参数中, 我们用 map 来存节点的值

在BFS遍历过程中 最先找到的与目标值相同的节点一定是最近的
因为是按层遍历 距离按层逐渐增加

public class Solution {
    public UndirectedGraphNode searchNode(ArrayList<UndirectedGraphNode> graph,
                                          Map<UndirectedGraphNode, Integer> values,
                                          UndirectedGraphNode node,
                                          int target) {
        if(node == null || values.get(node) == target)
            return node;
        Queue<UndirectedGraphNode> queue = new LinkedList<>();
        Set<UndirectedGraphNode> set = new HashSet<>();
        queue.offer(node);
        set.add(node);
        while(!queue.isEmpty()){
            UndirectedGraphNode n = queue.poll();
            for(UndirectedGraphNode neighbor : n.neighbors){
                if(set.contains(neighbor))
                    continue;
                if(values.get(neighbor) == target)
                    return neighbor;
                queue.offer(neighbor);
                set.add(neighbor);
            }
        }
        return null;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值