結果
問題 |
No.1370 置換門松列
|
ユーザー |
|
提出日時 | 2021-02-12 15:05:24 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,603 bytes |
コンパイル時間 | 225 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 110,592 KB |
最終ジャッジ日時 | 2024-11-08 04:19:09 |
合計ジャッジ時間 | 5,501 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 13 WA * 12 |
ソースコード
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+m<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が成り立つ必要がある。 # グラフに循環がなければOK # 循環をチェック # 入次数0の頂点が一つもないとき、グラフは循環している。 if len(v0ls)==0:return [] # 出次数0の頂点に大きい値を入れる。重複がないようにする。 ret=[-1]*m vl=[m] for v in range(m): if len(g[v])==0: ret[v]=vl[0] vl[0]-=1 mi=set(range(m)) seen=[0]*m def dfs(v):# 頂点vの値を決める。vから出ている辺があれば、その辺の先の頂点を先に決める。 dfsret=True if ret[v]!=-1:return True for nv in g[v]: if seen[nv]==1: return False seen[nv]=1 dfsret&=dfs(nv) seen[nv]=0 ret[v]=vl[0] vl[0]-=1 return dfsret lp=True for v0 in v0ls: lp&=dfs(v0) if not lp:return [] return ret if chk(n,m,a,ret): return [x-mnv+1 if x is not None else 1 for x in ret] else: return [] 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')