Tinh Hoa Đông Lào
Junior Member
Là contest 409 của tháng 8, tối qua vừa làm xong
Python:
class Solution:
def shortestDistanceAfterQueries(self, n: int, queries: List[List[int]]) -> List[int]:
graph = defaultdict(list)
for u, v in pairwise(range(n)):
graph[u].append(v)
def bfs(node, src):
q = deque([node])
visited = {node}
cnt = 0
while q:
cnt += 1
for _ in range(len(q)):
curr = q.popleft()
for nei in graph[curr]:
if nei not in visited:
q.append(nei)
visited.add(nei)
if nei == src:
return cnt
res = []
for u, v in queries:
graph[u].append(v)
res.append(bfs(0, n-1))
return res




đen vl