图论3(Leetcode841.钥匙和房间)

答案:

class Solution {
    public boolean canVisitAllRooms(List<List<Integer>> rooms) {
        Set<Integer> isArrive = new HashSet<>();
        Set<Integer> newArrive = new HashSet<>();
        isArrive.add(0);
        newArrive.add(0);
        while(newArrive.size()!=0){
            newArrive = getNewArrive(newArrive,isArrive,rooms);
            System.out.println(newArrive);

        }
        System.out.println(newArrive);
        if(isArrive.size()==rooms.size()){
            return true;
        }else{
            return false;
        }
    }
    public Set<Integer> getNewArrive(Set<Integer> newArrive,Set<Integer> isArrive,List<List<Integer>> rooms){
        Set<Integer> nextNewArrive = new HashSet<>();
        Iterator<Integer> iterator = newArrive.iterator();
        while (iterator.hasNext()) {//遍历newArrive
            int newRoom = iterator.next();
            for(int i=0;i<rooms.size();i++){//找到rooms里的newArrive元素的房间
                if(i==newRoom){
                    for(int k=0;k<rooms.get(i).size();k++){
                        int room = rooms.get(i).get(k);
                        if(!isArrive.contains(room)){
                            isArrive.add(room);
                            nextNewArrive.add(room);
                        }
                    }
                }

            }
        }
        return nextNewArrive;
    }
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值