結果

問題 No.1640 簡単な色塗り
ユーザー ygd.
提出日時 2021-08-06 22:21:42
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,747 bytes
コンパイル時間 241 ms
コンパイル使用メモリ 82,428 KB
実行使用メモリ 127,652 KB
最終ジャッジ日時 2024-06-29 15:23:45
合計ジャッジ時間 17,748 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 25 WA * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

from heapq import heapify, heappop, heappush
from collections import defaultdict 


def main():
    N = int(input())
    deg = [0]*N
    dic = defaultdict(list)
    for i in range(N):
        a,b = map(int,input().split())
        a -= 1; b -= 1
        deg[a] += 1
        deg[b] += 1
        dic[a].append((b,i)) #行先、辺の番号
        dic[b].append((a,i))
        
    PQ = []
    #for i in range(N):
    #    heappush(PQ,(deg[i],i))
    ret = []
    used_point = set([])
    used_edge = set([])
    for i in range(N):
        if i in used_point: continue
        heappush(PQ,(deg[i],i))
        while PQ:
            #print(PQ)
            d,v = heappop(PQ)
            #print(d,v)
            for u,edge_idx in dic[v]:
                if v in used_point and u in used_point: continue
                if edge_idx in used_edge: continue
                if v in used_point:
                    ret.append(u)
                    used_point.add(u)
                    used_edge.add(edge_idx)
                    deg[v] -= 1
                    deg[u] -= 1
                elif u in used_point:
                    ret.append(v)
                    used_point.add(v)
                    used_edge.add(edge_idx)
                    deg[v] -= 1
                    deg[u] -= 1
                else:
                    ret.append(v)
                    deg[v] -= 1
                    deg[u] -= 1
                    used_point.add(v)
                    used_edge.add(edge_idx)
                    #print("push",u,edge_idx)
                    heappush(PQ,(deg[u],u))
    if len(ret) == N:
        print('Yes')
        ret = [x+1 for x in ret]
        print(*ret,sep="\n")
    else:
        print('No')

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