結果
問題 |
No.1266 7 Colors
|
ユーザー |
![]() |
提出日時 | 2020-11-08 18:35:43 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 951 ms / 3,000 ms |
コード長 | 2,168 bytes |
コンパイル時間 | 255 ms |
コンパイル使用メモリ | 82,120 KB |
実行使用メモリ | 103,296 KB |
最終ジャッジ日時 | 2024-07-22 15:36:08 |
合計ジャッジ時間 | 14,818 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
import sys sys.setrecursionlimit(10**6) int1 = lambda x: int(x)-1 p2D = lambda x: print(*x, sep="\n") def II(): return int(sys.stdin.buffer.readline()) def MI(): return map(int, sys.stdin.buffer.readline().split()) def LI(): return list(map(int, sys.stdin.buffer.readline().split())) def LLI(rows_number): return [LI() for _ in range(rows_number)] def MI1(): return map(int1, sys.stdin.buffer.readline().split()) def LI1(): return list(map(int1, sys.stdin.buffer.readline().split())) def LLI1(rows_number): return [LI1() for _ in range(rows_number)] def BI(): return sys.stdin.buffer.readline().rstrip() def SI(): return sys.stdin.buffer.readline().rstrip().decode() class UnionFind: def __init__(self, n): self.state = [-1] * n self.size_table = [1] * n # cntはグループ数 # self.cnt = n def root(self, u): v = self.state[u] if v < 0: return u self.state[u] = res = self.root(v) return res def merge(self, u, v): ru = self.root(u) rv = self.root(v) if ru == rv: return du = self.state[ru] dv = self.state[rv] if du > dv: ru, rv = rv, ru if du == dv: self.state[ru] -= 1 self.state[rv] = ru # self.cnt -= 1 self.size_table[ru] += self.size_table[rv] return # グループの要素数 def size(self, u): return self.size_table[self.root(u)] n,m,q=MI() ss=[int(SI()[::-1],2) for _ in range(n)] to=[[] for _ in range(n)] uf=UnionFind(n*7) for _ in range(m): u,v=MI1() to[u].append(v) to[v].append(u) s=ss[u]&ss[v] for c in range(7): if s>>c&1==0:continue uf.merge(u*7+c,v*7+c) for u,s in enumerate(ss): if s&65==65:uf.merge(u*7,u*7+6) for i in range(6): if s&3==3:uf.merge(u*7+i,u*7+i+1) s>>=1 for _ in range(q): t,u,c=MI1() if t: print(uf.size(u*7)) else: ss[u]|=1<<c l=(c-1)%7 r=(c+1)%7 if ss[u]>>l&1:uf.merge(u*7+c,u*7+l) if ss[u]>>r&1:uf.merge(u*7+c,u*7+r) for v in to[u]: if ss[v]>>c&1==0:continue uf.merge(u*7+c,v*7+c)