結果
問題 | No.2674 k-Walk on Bipartite |
ユーザー |
![]() |
提出日時 | 2025-03-26 15:47:32 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,971 bytes |
コンパイル時間 | 161 ms |
コンパイル使用メモリ | 82,420 KB |
実行使用メモリ | 125,180 KB |
最終ジャッジ日時 | 2025-03-26 15:48:30 |
合計ジャッジ時間 | 6,574 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 WA * 12 |
ソースコード
import sysfrom collections import dequedef main():input = sys.stdin.read().split()ptr = 0N = int(input[ptr])ptr += 1M = int(input[ptr])ptr += 1s = int(input[ptr]) - 1 # 0-basedptr += 1t = int(input[ptr]) - 1ptr += 1k = int(input[ptr])ptr += 1edges = [[] for _ in range(N)]for _ in range(M):a = int(input[ptr]) - 1ptr += 1b = int(input[ptr]) - 1ptr += 1edges[a].append(b)edges[b].append(a)# Bipartition coloringcolor = [-1] * Nis_bipartite = Truefor start in range(N):if color[start] == -1:q = deque()q.append(start)color[start] = 0while q:u = q.popleft()for v in edges[u]:if color[v] == -1:color[v] = color[u] ^ 1q.append(v)elif color[v] == color[u]:is_bipartite = False# Since F is guaranteed to be bipartite, we don't need to handle non-bipartite cases_color = color[s]t_color = color[t]# Check parityif (s_color == t_color and k % 2 != 0) or (s_color != t_color and k % 2 != 1):print("No")return# Check connectivity and compute shortest distance d from s to tvisited = [False] * Ndist = [-1] * Nq = deque()q.append(s)dist[s] = 0visited[s] = Truefound = Falsewhile q:u = q.popleft()if u == t:found = Truebreakfor v in edges[u]:if not visited[v]:visited[v] = Truedist[v] = dist[u] + 1q.append(v)if not found:print("Unknown")returnd = dist[t]if d > k or (k - d) % 2 != 0:print("Unknown")else:print("Yes")if __name__ == "__main__":main()