結果

問題 No.2251 Marking Grid
コンテスト
ユーザー detteiuu
提出日時 2026-08-29 17:56:18
言語 PyPy3
(7.3.23 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 2,062 ms / 3,000 ms
+ 987µs
コード長 2,153 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 260 ms
コンパイル使用メモリ 96,104 KB
実行使用メモリ 258,388 KB
最終ジャッジ日時 2026-08-29 17:56:44
合計ジャッジ時間 20,925 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def op(x, y):
    return x|y
class UnionFind:
    def __init__(self, n, A, op):
        self.n = n
        self.parent_size = [-1]*n
        self.A = A[:]
        self.op = op
 
    def leader(self, a):
        if self.parent_size[a] < 0:
            return a
        self.parent_size[a] = self.leader(self.parent_size[a])
        return self.parent_size[a]
 
    def merge(self, a, b):
        x, y = self.leader(a), self.leader(b)
        if x == y:
            return 
        l, r = self.A[x], self.A[y]
        if abs(self.parent_size[x]) < abs(self.parent_size[y]):
            x, y = y, x
        self.parent_size[x] += self.parent_size[y]
        self.parent_size[y] = x
        self.A[x] = self.op(l, r)
        return 
    
    def __getitem__(self, n):
        return self.A[self.leader(n)]
    
    def update(self, n, a):
        self.A[self.leader(n)] = a
 
    def same(self, a, b):
        return self.leader(a) == self.leader(b)
 
    def size(self, a):
        return abs(self.parent_size[self.leader(a)])
 
    def groups(self):
        result = [[] for _ in range(self.n)]
        for i in range(self.n):
            result[self.leader(i)].append(i)
        return [r for r in result if r != []]

MOD = 998244353

H, W = map(int, input().split())
A = [list(map(int, input().split())) for _ in range(H)]

POW = [1]
for _ in range(H+W):
    POW.append(POW[-1]*2%MOD)

B = sorted([(i, j, A[i][j]) for i in range(H) for j in range(W)], key=lambda x:x[2], reverse=True)
rank = [[-1]*W for _ in range(H)]
IDX = [-1]*H
for idx, (i, j, a) in enumerate(B):
    rank[i][j] = idx
    if IDX[i] == -1: IDX[i] = j

FH = [False]*H
cntH = H
UF = [UnionFind(W, [0]*W, op) for _ in range(H)]
cnt = [W]*H
ans = 0
for i, j, a in B:
    if not FH[i]:
        FH[i] = True
        cntH -= 1
    if UF[i][j] == 0:
        cnt[i] -= 1
    flag = UF[i][j] == 0
    UF[i].update(j, 1)
    if flag:
        ans += a*POW[cntH+cnt[i]]%MOD
        ans %= MOD
    if IDX[i] != j and not UF[0].same(IDX[i], j):
        for k in range(H):
            if UF[k][IDX[i]] == 0 or UF[k][j] == 0:
                cnt[k] -= 1
            UF[k].merge(IDX[i], j)

print(ans)
0