結果

問題 No.1479 Matrix Eraser
ユーザー aaaaaaaaaa2230aaaaaaaaaa2230
提出日時 2021-04-16 23:00:57
言語 PyPy3
(7.3.13)
結果
RE  
実行時間 -
コード長 3,448 bytes
コンパイル時間 739 ms
コンパイル使用メモリ 87,092 KB
実行使用メモリ 282,200 KB
最終ジャッジ日時 2023-09-16 02:17:50
合計ジャッジ時間 27,118 ms
ジャッジサーバーID
(参考情報)
judge14 / judge13
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 158 ms
114,276 KB
testcase_01 AC 191 ms
136,080 KB
testcase_02 AC 188 ms
136,068 KB
testcase_03 AC 189 ms
136,124 KB
testcase_04 AC 191 ms
136,032 KB
testcase_05 AC 190 ms
136,008 KB
testcase_06 AC 192 ms
136,172 KB
testcase_07 AC 479 ms
155,084 KB
testcase_08 AC 576 ms
166,000 KB
testcase_09 AC 864 ms
195,372 KB
testcase_10 RE -
testcase_11 AC 954 ms
203,860 KB
testcase_12 RE -
testcase_13 RE -
testcase_14 AC 554 ms
161,304 KB
testcase_15 AC 327 ms
143,880 KB
testcase_16 RE -
testcase_17 AC 1,505 ms
252,932 KB
testcase_18 RE -
testcase_19 AC 1,523 ms
254,800 KB
testcase_20 AC 1,563 ms
254,380 KB
testcase_21 RE -
testcase_22 AC 1,532 ms
253,560 KB
testcase_23 RE -
testcase_24 RE -
testcase_25 AC 1,516 ms
253,260 KB
testcase_26 RE -
testcase_27 AC 616 ms
163,556 KB
testcase_28 AC 584 ms
163,656 KB
testcase_29 AC 563 ms
162,144 KB
testcase_30 AC 607 ms
163,660 KB
testcase_31 AC 572 ms
163,344 KB
testcase_32 AC 438 ms
192,808 KB
testcase_33 AC 422 ms
193,236 KB
testcase_34 AC 432 ms
193,020 KB
testcase_35 AC 436 ms
193,620 KB
testcase_36 AC 454 ms
193,156 KB
testcase_37 AC 193 ms
119,436 KB
testcase_38 AC 691 ms
199,028 KB
testcase_39 AC 1,555 ms
282,200 KB
testcase_40 AC 158 ms
114,076 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import deque
import sys
sys.setrecursionlimit(10**6)
class MaxFlow:
    inf = 10**18

    class E:
        def __init__(self,to,cap):
            self.to = to
            self.cap = cap
            self.rev = None

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

    def add_edge(self, fr, to, cap):
        graph = self.graph
        edge = self.E(to,cap)
        edge2 = self.E(fr,0)
        edge.rev = edge2
        edge2.rev = edge
        graph[fr].append(edge)
        graph[to].append(edge2)

    def bfs(self, s, t):
        level = self.level = [self.n]*self.n
        q = deque([s])
        level[s] = 0
        while q:
            now = q.popleft()
            lw = level[now]+1
            for e in self.graph[now]:
                if e.cap and level[e.to]> lw:
                    level[e.to] = lw
                    if e.to == t:
                        return True
                    q.append(e.to)
        return False

    def dfs(self, s, t, up):
        graph = self.graph
        it = self.it
        level = self.level

        st = deque([t])
        while st:
            v = st[-1]
            if v == s:
                st.pop()
                flow = up
                for w in st:
                    e = graph[w][it[w]].rev
                    flow = min(flow, e.cap)
                for w in st:
                    e = graph[w][it[w]]
                    e.cap += flow
                    e.rev.cap -= flow
                return flow
            lv = level[v]-1
            while it[v] < len(graph[v]):
                e = graph[v][it[v]]
                re = e.rev
                if re.cap == 0 or lv != level[e.to]:
                    it[v] += 1
                    continue
                st.append(e.to)
                break
            if it[v] == len(graph[v]):
                st.pop()
                level[v] = self.n

        return 0

    def flow(self,s,t,flow_limit=inf):
        flow = 0
        while flow < flow_limit and self.bfs(s,t):
            self.it = [0]*self.n
            while flow < flow_limit:
                f = self.dfs(s,t,flow_limit-flow)
                if f == 0:
                    break
                flow += f
        return flow

    def min_cut(self,s):
        visited = [0]*self.n
        q = deque([s])
        while q:
            v = q.pop()
            visited[v] = 1
            for e in self.graph[v]:
                if e.cap and not visited[e.to]:
                    q.append(e.to)
        return visited
h,w = map(int,input().split())
A = [list(map(int,input().split())) for i in range(h)]

xy = [[] for i in range(5*10**5+5)]
X = [set() for i in range(5*10**5)]
Y = [set() for i in range(5*10**5)]
for i in range(h):
    for j in range(w):
        num = A[i][j]
        if num == 0:
            continue
        X[num].add(i)
        Y[num].add(j)
        xy[num].append((i,j))

ans = 0
for i in range(5*10**5,0,-1):
    if xy[i] == []:
        continue
    nh = len(X[i])
    nw = len(Y[i])
    dich = {x:j for j,x in enumerate(sorted(list(X[i])))}
    dicw = {x:j for j,x in enumerate(sorted(list(Y[i])))}
    s = nh+nw
    t = s+1
    maxflow = MaxFlow(t+1)
    for c in X[i]:
        maxflow.add_edge(s, dich[c], 1)
    for c in Y[i]:
        maxflow.add_edge(nh+dicw[c], t, 1)
    for x,y in xy[i]:
        maxflow.add_edge(dich[x], nh+dicw[y], 1)
    ans += maxflow.flow(s, t)
print(ans)
0