結果
問題 | No.2563 色ごとのグループ |
ユーザー | とろちゃ |
提出日時 | 2023-12-02 15:39:45 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 818 ms / 2,000 ms |
コード長 | 1,993 bytes |
コンパイル時間 | 465 ms |
コンパイル使用メモリ | 81,700 KB |
実行使用メモリ | 199,704 KB |
最終ジャッジ日時 | 2023-12-02 15:40:01 |
合計ジャッジ時間 | 13,051 ms |
ジャッジサーバーID (参考情報) |
judge10 / judge9 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 42 ms
55,608 KB |
testcase_01 | AC | 42 ms
55,608 KB |
testcase_02 | AC | 41 ms
55,608 KB |
testcase_03 | AC | 41 ms
55,608 KB |
testcase_04 | AC | 50 ms
55,608 KB |
testcase_05 | AC | 75 ms
55,608 KB |
testcase_06 | AC | 46 ms
55,608 KB |
testcase_07 | AC | 46 ms
55,608 KB |
testcase_08 | AC | 43 ms
55,608 KB |
testcase_09 | AC | 43 ms
55,608 KB |
testcase_10 | AC | 42 ms
55,608 KB |
testcase_11 | AC | 44 ms
55,608 KB |
testcase_12 | AC | 43 ms
55,608 KB |
testcase_13 | AC | 43 ms
55,608 KB |
testcase_14 | AC | 93 ms
77,080 KB |
testcase_15 | AC | 85 ms
77,352 KB |
testcase_16 | AC | 81 ms
77,092 KB |
testcase_17 | AC | 74 ms
74,264 KB |
testcase_18 | AC | 63 ms
70,916 KB |
testcase_19 | AC | 89 ms
77,764 KB |
testcase_20 | AC | 134 ms
83,032 KB |
testcase_21 | AC | 119 ms
80,580 KB |
testcase_22 | AC | 104 ms
78,148 KB |
testcase_23 | AC | 121 ms
80,976 KB |
testcase_24 | AC | 441 ms
125,520 KB |
testcase_25 | AC | 368 ms
124,120 KB |
testcase_26 | AC | 498 ms
145,240 KB |
testcase_27 | AC | 482 ms
142,364 KB |
testcase_28 | AC | 576 ms
153,256 KB |
testcase_29 | AC | 792 ms
172,572 KB |
testcase_30 | AC | 763 ms
171,552 KB |
testcase_31 | AC | 748 ms
171,552 KB |
testcase_32 | AC | 818 ms
171,552 KB |
testcase_33 | AC | 746 ms
198,292 KB |
testcase_34 | AC | 764 ms
197,120 KB |
testcase_35 | AC | 789 ms
199,552 KB |
testcase_36 | AC | 737 ms
198,416 KB |
testcase_37 | AC | 756 ms
199,704 KB |
ソースコード
# import pypyjit;pypyjit.set_param("max_unroll_recursion=-1") import os import sys # from bisect import * from collections import * # from datetime import * # from decimal import * # PyPyだと遅い # from heapq import * # from itertools import * # from math import gcd, lcm # from random import * # from string import * # import numpy as np # from atcoder.dsu import * # from atcoder.segtree import * # from more_itertools import * # from sortedcontainers import * # sys.setrecursionlimit(10**7) # PyPyは呪文を付ける # sys.set_int_max_str_digits(0) INF = 1 << 61 MOD = 998244353 # MOD = 10**9 + 7 IS_ATCODER = os.getenv("ATCODER", 0) File = sys.stdin def input(): return File.readline().replace("\n", "") # /////////////////////////////////////////////////////////////////////////// def bfs(G, start): """ グラフで幅優先探索をする関数 seenを集合、distを辞書で管理 配列で管理するより遅め 引数: G (defaultdict): グラフ start (int): 始点 戻り値: dist (dict): 始点からの距離 """ que = deque() dist = {} global seen que.append(start) seen.add(start) dist[start] = 0 while que: for _ in range(len(que)): pop = que.popleft() for i in G[pop]: if i not in seen and C[i] == C[pop]: seen.add(i) dist[i] = dist.get(pop, 0) + 1 que.append(i) return dist N, M = map(int, input().split()) C = list(map(int, input().split())) edges = [list(map(int, input().split())) for _ in range(M)] G = [set() for _ in range(N)] for i, j in edges: G[i - 1].add(j - 1) G[j - 1].add(i - 1) color = defaultdict(list) for i, j in enumerate(C): color[j].append(i) ans = 0 for i in color.keys(): seen = set() c = 0 for j in color[i]: if j in seen: continue c += 1 bfs(G, j) ans += c - 1 print(ans)