結果

問題 No.1370 置換門松列
コンテスト
ユーザー persimmon-persimmon
提出日時 2021-02-12 15:29:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 183 ms / 2,000 ms
コード長 1,806 bytes
コンパイル時間 178 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 110,592 KB
最終ジャッジ日時 2024-09-14 19:25:09
合計ジャッジ時間 3,554 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10**7)
def main1(n,m,a):
  # 自明な例を除外
  for i in range(n-2):
    if a[i]!=a[i+1]!=a[i+2]!=a[i]:
      continue
    return []

  g=[[] for _ in range(m)]
  v0ls=set(range(m))
  for i in range(0,n,2):
    if 0<=i-1:
      u,v=a[i],a[i-1]
      u,v=u-1,v-1
      g[u].append(v)
      v0ls.discard(v)
    if i+1<n:
      u,v=a[i],a[i+1]
      u,v=u-1,v-1
      g[u].append(v)
      v0ls.discard(v)
  
  # u->vに辺がある<=>u<vが成り立つ必要がある。
  # 大きい値から設定することを考える。
  # 頂点vに次の値に設定するとき、vから延びる辺の先の頂点にはすべて値が設定されている。
  # グラフに循環がなければOK
  
  # 入次数0の頂点が一つもないとき、グラフは循環している。
  if len(v0ls)==0:return []
  
  ret=[-1]*m
  vl=[m]
  seen=[0]*m
  # 頂点vの値を決める。vから出ている辺があれば、その辺の先の頂点を先に決める。
  # たどった先に循環があればFalseを返す。なければTrueを返す。
  def dfs(v):
    # 頂点に値が設定されているなら、その先の頂点にもすべて値が設定されている。
    if ret[v]!=-1:return True

    for nv in g[v]:
      # 循環している。
      if seen[nv]==1:return False
      seen[nv]=1
      if not dfs(nv):return False
      seen[nv]=0
    # 先の頂点にすべて値を設定した後、vに値を設定する。
    ret[v]=vl[0]
    vl[0]-=1
    return True
  for v0 in range(m):
    seen[v0]=1
    lp=dfs(v0)
    seen[v0]=0
    if not lp:return []
  return ret
if __name__=='__main__':
  n,m=map(int,input().split())
  a=list(map(int,input().split()))
  ret1=main1(n,m,a)
  if ret1:
    print('Yes')
    print(*ret1)
  else:
    print('No')
0