結果

問題 No.1479 Matrix Eraser
ユーザー 草苺奶昔草苺奶昔
提出日時 2023-03-14 01:50:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 406 ms / 3,000 ms
コード長 3,623 bytes
コンパイル時間 231 ms
コンパイル使用メモリ 82,280 KB
実行使用メモリ 136,504 KB
最終ジャッジ日時 2024-09-18 07:51:46
合計ジャッジ時間 11,807 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 60 ms
68,456 KB
testcase_01 AC 62 ms
69,736 KB
testcase_02 AC 62 ms
68,512 KB
testcase_03 AC 58 ms
68,644 KB
testcase_04 AC 60 ms
68,556 KB
testcase_05 AC 60 ms
68,152 KB
testcase_06 AC 61 ms
68,372 KB
testcase_07 AC 156 ms
86,284 KB
testcase_08 AC 177 ms
92,336 KB
testcase_09 AC 237 ms
103,128 KB
testcase_10 AC 352 ms
119,392 KB
testcase_11 AC 265 ms
109,272 KB
testcase_12 AC 152 ms
88,820 KB
testcase_13 AC 180 ms
92,468 KB
testcase_14 AC 170 ms
88,760 KB
testcase_15 AC 112 ms
80,036 KB
testcase_16 AC 176 ms
90,172 KB
testcase_17 AC 379 ms
131,968 KB
testcase_18 AC 400 ms
131,972 KB
testcase_19 AC 395 ms
132,752 KB
testcase_20 AC 381 ms
132,260 KB
testcase_21 AC 386 ms
132,192 KB
testcase_22 AC 366 ms
132,152 KB
testcase_23 AC 396 ms
132,612 KB
testcase_24 AC 406 ms
132,220 KB
testcase_25 AC 378 ms
132,188 KB
testcase_26 AC 377 ms
132,356 KB
testcase_27 AC 209 ms
91,588 KB
testcase_28 AC 205 ms
91,784 KB
testcase_29 AC 216 ms
91,784 KB
testcase_30 AC 220 ms
91,660 KB
testcase_31 AC 211 ms
91,792 KB
testcase_32 AC 203 ms
125,156 KB
testcase_33 AC 204 ms
124,888 KB
testcase_34 AC 193 ms
125,004 KB
testcase_35 AC 215 ms
133,600 KB
testcase_36 AC 196 ms
124,764 KB
testcase_37 AC 106 ms
97,040 KB
testcase_38 AC 180 ms
90,556 KB
testcase_39 AC 330 ms
136,504 KB
testcase_40 AC 60 ms
69,312 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
from typing import List
from collections import deque
from typing import List, Tuple


class Hungarian:
    """
    軽量化Dinic法
    ref : https://snuke.hatenablog.com/entry/2019/05/07/013609
    """

    @staticmethod
    def fromEdges(n: int, edges: List[Tuple[int, int]]) -> "Hungarian":
        """从无向边列表构造"""
        adjList = [[] for _ in range(n)]
        for u, v in edges:
            adjList[u].append(v)
            adjList[v].append(u)
        colors, ok = isBipartite(n, adjList)
        if not ok:
            raise ValueError("not bipartite")
        h = Hungarian(n)
        for u, v in edges:
            if colors[u] == 1:
                u, v = v, u
            h.addEdge(u, v)
        return h

    def __init__(self, n: int):
        self._n = n
        self._to = [[] for _ in range(n)]

    def addEdge(self, u: int, v: int) -> None:
        """男孩u和女孩v连边"""
        self._to[u].append(v)

    def work(self) -> List[Tuple[int, int]]:
        """返回最大匹配"""
        n, to = self._n, self._to
        pre = [-1] * n
        root = [-1] * n
        p = [-1] * n
        q = [-1] * n
        upd = True
        while upd:
            upd = False
            s = []
            s_front = 0
            for i in range(n):
                if p[i] == -1:
                    root[i] = i
                    s.append(i)
            while s_front < len(s):
                v = s[s_front]
                s_front += 1
                if p[root[v]] != -1:
                    continue
                for u in to[v]:
                    if q[u] == -1:
                        while u != -1:
                            q[u] = v
                            p[v], u = u, p[v]
                            v = pre[v]
                        upd = True
                        break
                    u = q[u]
                    if pre[u] != -1:
                        continue
                    pre[u] = v
                    root[u] = root[v]
                    s.append(u)
            if upd:
                for i in range(n):
                    pre[i] = -1
                    root[i] = -1
        return [(v, p[v]) for v in range(n) if p[v] != -1]


def isBipartite(n: int, adjList: List[List[int]]) -> Tuple[List[int], bool]:
    """二分图检测 bfs染色"""

    def bfs(start: int) -> bool:
        colors[start] = 0
        queue = deque([start])
        while queue:
            cur = queue.popleft()
            for next in adjList[cur]:
                if colors[next] == -1:
                    colors[next] = colors[cur] ^ 1
                    queue.append(next)
                elif colors[next] == colors[cur]:
                    return False
        return True

    colors = [-1] * n
    for i in range(n):
        if colors[i] == -1 and not bfs(i):
            return [], False
    return colors, True



def solve(grid: List[List[int]]) -> int:
    ROW, COL = len(grid), len(grid[0])
    mp = defaultdict(list)
    for i in range(ROW):
        for j in range(COL):
            mp[grid[i][j]].append((i, j))

    res = 0
    for v, edges in mp.items():
        if v == 0:
            continue
        H = Hungarian(len(edges))
        id1, id2 = dict(), dict()
        for u, v in edges:
            id1.setdefault(u, len(id1))
            id2.setdefault(v, len(id2))
            H.addEdge(id1[u], id2[v])
        res += len(H.work())

    return res


if __name__ == "__main__":
    n, m = map(int, input().split())
    grid = [list(map(int, input().split())) for _ in range(n)]
    print(solve(grid))
0