結果

問題 No.2674 k-Walk on Bipartite
ユーザー ゼット
提出日時 2024-03-15 22:50:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 484 ms / 2,000 ms
コード長 1,881 bytes
コンパイル時間 131 ms
コンパイル使用メモリ 82,720 KB
実行使用メモリ 127,140 KB
最終ジャッジ日時 2024-09-30 02:20:35
合計ジャッジ時間 7,338 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

class unif:
  def __init__(self,n):
    self.pare=[-1]*n
    self.size=[1]*n
  def root(self,x):
    while self.pare[x]!=-1:
      x=self.pare[x]
    return x
  def unite(self,u,v):
    rootu=self.root(u)
    rootv=self.root(v)
    if rootu!=rootv:
      if self.size[rootu]>=self.size[rootv]:
        self.pare[rootv]=rootu
        self.size[rootu]+=self.size[rootv]
      else:
        self.pare[rootu]=rootv
        self.size[rootv]+=self.size[rootu]
  def same(self,s,t):
    return self.root(s)==self.root(t)
N,M=map(int,input().split())
s,t,K=map(int,input().split())
if N==1:
  print('No')
  exit()
if N==2:
  if s==t:
    if K%2==0:
      if M==1:
        print('Yes')
      else:
        print('Unknown')
    else:
      print('No')
  else:
    if K%2==0:
      print('No')
    else:
      if M==1:
        print('Yes')
      else:
        print('Unknown')
  exit()
G=[[] for i in range(N)]
Z=unif(N)
for i in range(M):
  a,b=map(int,input().split())
  G[a-1].append(b-1)
  G[b-1].append(a-1)
  Z.unite(a-1,b-1)
if Z.same(s-1,t-1)==False:
  print('Unknown')
  exit()
dist=[-1]*N
from collections import deque
S=deque()
p=[[0]*N for i in range(20)]
for i in range(N):
  if dist[i]>=0:
    continue
  dist[i]=0
  S.append(i)
  p[0][i]=i
  while S:
    x=S.popleft()
    for y in G[x]:
      if dist[y]>=0:
        continue
      dist[y]=dist[x]+1
      p[0][y]=x
      S.append(y)
for k in range(1,20):
  for i in range(N):
    p[k][i]=p[k-1][p[k-1][i]]
def LCA(x,y):
  a,b=x,y
  if dist[a]>dist[b]:
    a,b=b,a
  e=dist[b]-dist[a]
  for k in range(20):
    if (e>>k)&1:
      b=p[k][b]
  if a==b:
    return a
  for k in range(19,-1,-1):
    if p[k][a]!=p[k][b]:
      a=p[k][a]
      b=p[k][b]
  a=p[0][a]
  return a
s-=1
t-=1
if K%2!=(dist[s]-dist[t])%2:
  print('No')
  exit()
pos=LCA(s,t)
d=dist[s]+dist[t]-2*dist[pos]
if d>K:
  print('Unknown')
else:
  print('Yes')
0