結果

問題 No.1266 7 Colors
コンテスト
ユーザー c-yan
提出日時 2020-10-23 23:46:16
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,421 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 176 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 66,688 KB
最終ジャッジ日時 2024-07-21 13:47:14
合計ジャッジ時間 21,991 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 4 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from sys import setrecursionlimit, stdin
def find(parent, i):
    t = parent[i]
    if t < 0:
        return i
    t = find(parent, t)
    parent[i] = t
    return t


def unite(parent, i, j):
    i = find(parent, i)
    j = find(parent, j)
    if i == j:
        return
    parent[j] += parent[i]
    parent[i] = j
readline = stdin.readline
setrecursionlimit(10 ** 6)

N, M, Q = map(int, readline().split())
s = [readline() for _ in range(N)]
c = [tuple(map(int, x[:-1])) for x in s]

parent = [-1] * N

for _ in range(M):
    u, v = map(int, readline().split())
    u, v = u - 1, v - 1
    up = find(parent, u)
    vp = find(parent, v)
    if up != vp:
        x = tuple(c[up][i] + c[vp][i] for i in range(7))
        c[up] = x
        c[vp] = x
        unite(parent, u, v)

result = []
for _ in range(Q):
    t, x, y = map(int, readline().split())
    if t == 1:
        x, y = x - 1, y - 1
        xp = find(parent, x)
        z = 1 << (6 - y)
        a = list(c[xp])
        a[y] += 1
        c[xp] = tuple(a)
    elif t == 2:
        x = x - 1
        xp = find(parent, x)
        a = set()
        b = c[xp]
        for i in range(7):
            if b[i] == 0:
                break
            a.add(i)
        for i in range(6, -1, -1):
            if b[i] == 0:
                break
            a.add(i)
        d = 0
        for i in a:
            d += b[i]
        result.append(d)
print(*result, sep='\n')
0