結果
問題 | No.1266 7 Colors |
ユーザー |
![]() |
提出日時 | 2020-12-29 16:52:58 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 918 ms / 3,000 ms |
コード長 | 1,817 bytes |
コンパイル時間 | 286 ms |
コンパイル使用メモリ | 81,776 KB |
実行使用メモリ | 98,920 KB |
最終ジャッジ日時 | 2024-10-05 18:20:21 |
合計ジャッジ時間 | 13,738 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
import sysinput = sys.stdin.readlinesys.setrecursionlimit(10 ** 7)class UF_tree:def __init__(self, n):""" 0-indexed """self.root = [-1] * nself.components = ndef find(self, x):if self.root[x] < 0:return xself.root[x] = self.find(self.root[x])return self.root[x]def same(self, x, y):return self.find(x) == self.find(y)def merge(self, x, y):x = self.find(x)y = self.find(y)if x == y:return Falseif -self.root[x] > -self.root[y]:x, y = y, xself.root[y] += self.root[x]self.root[x] = yself.components -= 1return Truedef size(self, x):return -self.root[self.find(x)]N, M, Q = map(int, input().split())uf = UF_tree(7 * N)S = [sum(1 << i for i, s in enumerate(input().rstrip()) if s == "1")for _ in range(N)]for i, s in enumerate(S):i *= 7for j in range(7):k = (j + 1) % 7if (s >> j) & 1 and (s >> k) & 1:uf.merge(i + j, i + k)edge = [[] for _ in range(N)]for _ in range(M):a, b = map(int, input().split())a -= 1b -= 1edge[a].append(b)edge[b].append(a)for i in range(7):if (S[a] >> i) & 1 and (S[b] >> i) & 1:uf.merge(7 * a + i, 7 * b + i)for _ in range(Q):f, x, y = map(int, input().split())x -= 1y -= 1if f == 1:S[x] |= (1 << y)l = (y - 1) % 7if (S[x] >> l) & 1:uf.merge(7 * x + y, 7 * x + l)r = (y + 1) % 7if (S[x] >> r) & 1:uf.merge(7 * x + y, 7 * x + r)for z in edge[x]:if (S[z] >> y) & 1:uf.merge(7 * x + y, 7 * z + y)else:print(uf.size(7 * x))