結果
問題 | No.2563 色ごとのグループ |
ユーザー | kinoko_econ |
提出日時 | 2023-12-02 15:36:23 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,030 ms / 2,000 ms |
コード長 | 1,546 bytes |
コンパイル時間 | 533 ms |
コンパイル使用メモリ | 81,700 KB |
実行使用メモリ | 198,136 KB |
最終ジャッジ日時 | 2023-12-02 15:36:36 |
合計ジャッジ時間 | 10,964 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge10 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 36 ms
53,460 KB |
testcase_01 | AC | 36 ms
53,460 KB |
testcase_02 | AC | 35 ms
53,460 KB |
testcase_03 | AC | 37 ms
53,460 KB |
testcase_04 | AC | 35 ms
53,460 KB |
testcase_05 | AC | 36 ms
53,460 KB |
testcase_06 | AC | 36 ms
53,460 KB |
testcase_07 | AC | 36 ms
53,460 KB |
testcase_08 | AC | 37 ms
53,460 KB |
testcase_09 | AC | 37 ms
53,460 KB |
testcase_10 | AC | 35 ms
53,460 KB |
testcase_11 | AC | 38 ms
53,460 KB |
testcase_12 | AC | 37 ms
53,460 KB |
testcase_13 | AC | 36 ms
53,460 KB |
testcase_14 | AC | 72 ms
73,504 KB |
testcase_15 | AC | 80 ms
77,092 KB |
testcase_16 | AC | 72 ms
73,768 KB |
testcase_17 | AC | 69 ms
73,248 KB |
testcase_18 | AC | 58 ms
68,712 KB |
testcase_19 | AC | 83 ms
76,408 KB |
testcase_20 | AC | 114 ms
82,268 KB |
testcase_21 | AC | 99 ms
76,664 KB |
testcase_22 | AC | 100 ms
77,176 KB |
testcase_23 | AC | 101 ms
77,048 KB |
testcase_24 | AC | 231 ms
101,656 KB |
testcase_25 | AC | 222 ms
105,580 KB |
testcase_26 | AC | 290 ms
119,748 KB |
testcase_27 | AC | 298 ms
136,752 KB |
testcase_28 | AC | 311 ms
124,540 KB |
testcase_29 | AC | 382 ms
140,356 KB |
testcase_30 | AC | 392 ms
140,360 KB |
testcase_31 | AC | 376 ms
140,232 KB |
testcase_32 | AC | 382 ms
140,364 KB |
testcase_33 | AC | 1,030 ms
198,136 KB |
testcase_34 | AC | 845 ms
184,972 KB |
testcase_35 | AC | 935 ms
186,784 KB |
testcase_36 | AC | 995 ms
195,940 KB |
testcase_37 | AC | 826 ms
187,600 KB |
ソースコード
import sys sys.setrecursionlimit(10**6) N, M = map(int,input().split()) C = list(map(int,input().split())) G = [[] for _ in range(N)] for i in range(M): u, v = map(int,input().split()) if C[u-1] == C[v-1]: G[u-1].append(v-1) G[v-1].append(u-1) group = [[] for _ in range(N)] for i in range(N): group[C[i]-1].append(i) set_group = [set(group[i]) for i in range(N)] lst = [0]*N for i in range(N): for j in range(len(group[i])): lst[group[i][j]] = j def dfs(j, G, seen, i): # 頂点 v を探索済みにする seen[lst[group[i][j]]] = True # G[v] に含まれる頂点 v2 について、 for v2 in G[group[i][j]]: # v2 がすでに探索済みならば、スキップする if seen[lst[v2]]: continue # v2 始点で深さ優先探索を行う (関数を再帰させる) dfs(lst[v2], G, seen, i) return Ans = 0 for i in range(N): if len(group[i]) <= 1: continue seen = [False for _ in range(len(group[i]))] # seen[v]:頂点 v が探索済みなら true, そうでないなら false ans = 0 # 答えとなる変数 # 全ての頂点について for j in range(len(group[i])): # 頂点 v がすでに訪問済みであれば、スキップ if seen[lst[group[i][j]]] : continue # そうでなければ、頂点 v を含む連結成分は未探索 # 深さ優先探索で新たに訪問し、答えを 1 増やす dfs(j, G, seen, i) ans += 1 Ans += ans-1 print(Ans)