結果

問題 No.1640 簡単な色塗り
ユーザー ygd.
提出日時 2021-08-07 22:47:25
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 275 ms / 2,000 ms
コード長 2,279 bytes
コンパイル時間 199 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 117,120 KB
最終ジャッジ日時 2024-06-29 17:01:21
合計ジャッジ時間 16,981 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 53
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    N = int(input()); INF = pow(10,20)
    deg = [0]*N
    G = [[] for _ in range(N)]
    Z = []
    for i in range(N):
        a,b = map(int,input().split())
        a -= 1; b -= 1
        deg[a] += 1
        deg[b] += 1
        G[a].append((b,i)) #頂点、辺の番号
        G[b].append((a,i))
        Z.append((a,b))
 
    Q = []
    for i in range(N):
        if deg[i] == 1: #繋がっている辺が1つのものをリストへ
            Q.append(i)

    ans = [-1]*N #辺ごとにどちらを選んだか。
    used = set([])

    #残っている辺が1つの時、その頂点と辺を選択して、もう一つの頂点の次数が1なら追加。
    while Q:
        v = Q.pop()
        used.add(v)
        for u, idx in G[v]:
            if u in used: continue
            ans[idx] = v
            deg[v] -= 1
            deg[u] -= 1
            if deg[u] == 1:
                Q.append(u)
    #print(used)
    #ループや自己辺になっていて次数1が残っていないものを順番に調べる。
    for i in range(N):
        if i not in used:
            v = i
            while True:
                Flag = 0
                for u, idx in G[v]:
                    if u in used: continue
                    ans[idx] = v
                    used.add(v)
                    deg[v] -= 1; deg[u] -= 1
                    if v != u:
                        v = u #ここで入れ替える
                        Flag = 1
                    #自己辺の場合は個の頂点はもう終わり
                    break
                #print("Flag",Flag,"v",v)
                #print(ans,used)
                if Flag == 0: #最後の1頂点は残る。
                    break
    #print(ans,used)
    for idx in range(N): #使われていない辺をチェック
        if ans[idx] != -1: continue #既に使われた辺
        a,b = Z[idx]
        if a not in used:
            ans[idx] = a
            used.add(a)
        elif b not in used:
            ans[idx] = b
            used.add(b)
        assert all([a in used, b in used])

    #print(ans)
    if -1 not in ans:
        print('Yes')
        ret = [x+1 for x in ans]
        print(*ret,sep="\n")
    else:
        print('No')

if __name__ == '__main__':
    main()
0