結果

問題 No.1479 Matrix Eraser
ユーザー 小野寺健
提出日時 2021-04-28 12:56:53
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,383 bytes
コンパイル時間 92 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 72,068 KB
最終ジャッジ日時 2024-07-07 10:30:35
合計ジャッジ時間 54,017 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28 WA * 1 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict

def dfs(v, visited):
    global edges, matched
    for u in edges[v]:
        if u in visited:
            continue
        visited.add(u)
        if matched[u] == -1 or dfs(matched[u], visited) >= 0:
            matched[u] = v
            return u
    return -1
    
def mpc(xn, yn):
    global matched, edges
    matched = [-1] * yn

    left = [True] * xn
    right = [False] * yn

    ledges = []
    redges = [-1] * yn

    for v in range(xn):
        u = dfs(v, set()) 
        if u >= 0:
            redges[u] = v
        else:
            left[v] = False
            ledges.append(v)
    for v in ledges:
        for u in edges[v]:
            right[u] = True
            if redges[u] >= 0:
                left[redges[u]] = False

    return sum(left+right)


if __name__ == '__main__':
    H, W = map(int, input().split())

    S = defaultdict(list)

    for i in range(H):
        for j, a in enumerate(list(map(int, input().split()))):
            if a > 0:
                S[a].append((i, j))
    
    cnt = 0
    for a, l in S.items():
        h , w = set(), set()
        for i, j in l:
            h.add(i)
            w.add(j)
        h, w = list(h), list(w)
        edges = [set() for _ in range(len(h))]
        for i, j in l:
            edges[h.index(i)].add(w.index(j))
        cnt += mpc(len(h), len(w))
    
    print(cnt)
0