多益网络2019校招—钥匙和房间问题

时间：20180906

题目描述 ：

LeetCode 841. Keys and Rooms

有 N 个房间，开始时你位于 0 号房间。每个房间有不同的号码：0，1，2，...，N-1，并且房间里可能有一些钥匙能使你进入下一个房间。

在形式上，对于每个房间 i 都有一个钥匙列表 rooms[i]，每个钥匙 rooms[i][j] 由 [0,1，...，N-1] 中的一个整数表示， 其中 N = rooms.length。 钥匙 rooms[i][j] = v 可以打开编号为 v 的房间。 

最初，除 0 号房间外的其余所有房间都被锁住。你可以自由地在房间之间来回走动。如果能进入每个房间返回 true，否则返回 false。要求先写出解题思路，再编写代码。

通过广度优先搜索（BFS）的方式进行探索。将可访问的0房间加入队列之中，访问队列中的一个元素，遍历他可以到达的房间，如果该房间还不能被访问，那么设置为访问，否则跳过。

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Stack;

public class KeysAndRooms {
	public static void main(String[] args) {
		//[[1],[2],[3],[]]
		List<List<Integer>> rooms = new ArrayList();
		List<Integer> keys0 = new ArrayList<>();
		keys0.add(1);
		List<Integer> keys1 = new ArrayList<>();
		keys1.add(2);
		List<Integer> keys2 = new ArrayList<>();
		keys2.add(3);
		List<Integer> keys3 = new ArrayList<>();
		
		rooms.add(keys0);
		rooms.add(keys1);
		rooms.add(keys2);
		rooms.add(keys3);
		
		System.out.println(canVisitAllRooms(rooms));
	}
    private static boolean canVisitAllRooms(List<List<Integer>> rooms) {
		//用栈存储下一步要参观的房间
		Stack<Integer> dfs = new Stack<>(); 
		dfs.add(0);
		//用hashset存储已经参观过的房间
        HashSet<Integer> seen = new HashSet<Integer>(); 
        seen.add(0);
        while (!dfs.isEmpty()) {
            int i = dfs.pop();
            for (int j : rooms.get(i))
                if (!seen.contains(j)) {
                    dfs.add(j);
                    seen.add(j);
                    if (rooms.size() == seen.size()) return true;
                }
        }
        return rooms.size() == seen.size();
	}
	
}