結果
問題 | No.1266 7 Colors |
ユーザー | nagiss |
提出日時 | 2020-10-24 00:18:07 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 1,696 ms / 3,000 ms |
コード長 | 2,336 bytes |
コンパイル時間 | 101 ms |
コンパイル使用メモリ | 11,268 KB |
実行使用メモリ | 65,248 KB |
最終ジャッジ日時 | 2023-09-28 19:31:09 |
合計ジャッジ時間 | 23,450 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 19 ms
8,584 KB |
testcase_01 | AC | 18 ms
8,424 KB |
testcase_02 | AC | 17 ms
8,524 KB |
testcase_03 | AC | 929 ms
28,572 KB |
testcase_04 | AC | 1,152 ms
53,720 KB |
testcase_05 | AC | 961 ms
30,592 KB |
testcase_06 | AC | 1,198 ms
58,172 KB |
testcase_07 | AC | 1,271 ms
64,088 KB |
testcase_08 | AC | 1,164 ms
54,952 KB |
testcase_09 | AC | 1,042 ms
45,860 KB |
testcase_10 | AC | 1,016 ms
42,648 KB |
testcase_11 | AC | 932 ms
33,404 KB |
testcase_12 | AC | 960 ms
36,384 KB |
testcase_13 | AC | 983 ms
39,200 KB |
testcase_14 | AC | 940 ms
30,444 KB |
testcase_15 | AC | 1,277 ms
65,248 KB |
testcase_16 | AC | 985 ms
37,156 KB |
testcase_17 | AC | 1,261 ms
62,600 KB |
testcase_18 | AC | 1,678 ms
61,716 KB |
testcase_19 | AC | 1,696 ms
64,096 KB |
testcase_20 | AC | 1,694 ms
64,044 KB |
testcase_21 | AC | 293 ms
21,500 KB |
ソースコード
import sys class UnionFind: def __init__(self, N): self.p = list(range(N)) self.rank = [0] * N self.size = [1] * N def root(self, x): if self.p[x] != x: self.p[x] = self.root(self.p[x]) return self.p[x] def same(self, x, y): return self.root(x) == self.root(y) def unite(self, x, y): u = self.root(x) v = self.root(y) if u == v: return if self.rank[u] < self.rank[v]: self.p[u] = v self.size[v] += self.size[u] self.size[u] = 0 else: self.p[v] = u self.size[u] += self.size[v] self.size[v] = 0 if self.rank[u] == self.rank[v]: self.rank[u] += 1 def count(self, x): return self.size[self.root(x)] def main(): input = sys.stdin.buffer.readline N, M, Q = map(int, input().split()) S = [int(input(), 2) for _ in range(N)] G = [[] for _ in range(N)] for i in range(M): u, v = map(int, input().split()) u -= 1 v -= 1 G[u].append(v) G[v].append(u) uf = UnionFind(7 * N) for v, s in enumerate(S): ones = [i for i in range(7) if s>>(6-i)&1] v_offset = 7 * v if len(ones) >= 2: for o1, o2 in zip(ones, ones[1:]): if o1 + 1 == o2: uf.unite(v_offset+o1, v_offset+o2) if ones[0] == 0 and ones[-1] == 6: uf.unite(v_offset, v_offset+6) for u in G[v]: for o in ones: if S[u]>>(6-o)&1: uf.unite(v_offset+o, u*7+o) Ans = [] for _ in range(Q): t, x, y = map(int, input().split()) x -= 1 if t == 1: v_offset = x * 7 d = 7-y S[x] |= 1 << d v = v_offset + (y-1) d2 = (d-1)%7 if S[x] >> d2 & 1: uf.unite(v, v_offset+y%7) d2 = (d+1)%7 if S[x] >> d2 & 1: uf.unite(v, v_offset+(y-2)%7) for u in G[x]: if S[u]>>d&1: u = u * 7 + y-1 uf.unite(v, u) else: ans = uf.count(x*7) Ans.append(ans) print("\n".join(map(str, Ans))) main()