Medium
Paths in Maze That Lead to Same Room — Python
Full explanation · Time O(\ · Space V\
# Time: O(|V|^3)
# Space: O(|E|)
class Solution(object):
def numberOfPaths(self, n, corridors):
"""
:type n: int
:type corridors: List[List[int]]
:rtype: int
"""
adj = [set() for _ in xrange(n)]
for u, v in corridors:
adj[min(u, v)-1].add(max(u, v)-1)
return sum(k in adj[i] for i in xrange(n) for j in adj[i] for k in adj[j])