結果
問題 | No.1370 置換門松列 |
ユーザー |
|
提出日時 | 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 |
ソースコード
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')