結果
問題 | No.1479 Matrix Eraser |
ユーザー |
![]() |
提出日時 | 2021-04-16 22:21:05 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,040 ms / 3,000 ms |
コード長 | 6,210 bytes |
コンパイル時間 | 1,198 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 146,328 KB |
最終ジャッジ日時 | 2024-07-03 02:20:25 |
合計ジャッジ時間 | 20,218 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 39 |
ソースコード
import sys# sys.setrecursionlimit(10**6)int1 = lambda x: int(x)-1p2D = lambda x: print(*x, sep="\n")def II(): return int(sys.stdin.buffer.readline())def FI(): return float(sys.stdin.buffer.readline())def LI(): return list(map(int, sys.stdin.buffer.readline().split()))def LI1(): return list(map(int1, sys.stdin.buffer.readline().split()))def LLI(rows_number): return [LI() for _ in range(rows_number)]def LLI1(rows_number): return [LI1() for _ in range(rows_number)]def BI(): return sys.stdin.buffer.readline().rstrip()def SI(): return sys.stdin.buffer.readline().rstrip().decode()# dij = [(0, 1), (-1, 0), (0, -1), (1, 0)]# dij = [(0, 1), (-1, 0), (0, -1), (1, 0), (1, 1), (1, -1), (-1, 1), (-1, -1)]dij = [(0, 1), (1, 0), (1, 1), (1, -1)]inf = 10**16md = 998244353# md = 10**9+7from collections import defaultdictfrom typing import NamedTuple, Optional, List, castclass MFGraph:class Edge(NamedTuple):src: intdst: intcap: intflow: intclass _Edge:def __init__(self, dst: int, cap: int) -> None:self.dst = dstself.cap = capself.rev: Optional[MFGraph._Edge] = Nonedef __init__(self, n: int) -> None:self._n = nself._g: List[List[MFGraph._Edge]] = [[] for _ in range(n)]self._edges: List[MFGraph._Edge] = []def add_edge(self, src: int, dst: int, cap: int) -> int:assert 0 <= src < self._nassert 0 <= dst < self._nassert 0 <= capm = len(self._edges)e = MFGraph._Edge(dst, cap)re = MFGraph._Edge(src, 0)e.rev = rere.rev = eself._g[src].append(e)self._g[dst].append(re)self._edges.append(e)return mdef add_undir_edge(self, src: int, dst: int, cap: int) -> int:assert 0 <= src < self._nassert 0 <= dst < self._nassert 0 <= capm = len(self._edges)e = MFGraph._Edge(dst, cap)re = MFGraph._Edge(src, cap)e.rev = rere.rev = eself._g[src].append(e)self._g[dst].append(re)self._edges.append(e)return mdef get_edge(self, i: int) -> Edge:assert 0 <= i < len(self._edges)e = self._edges[i]re = cast(MFGraph._Edge, e.rev)return MFGraph.Edge(re.dst,e.dst,e.cap + re.cap,re.cap)def edges(self) -> List[Edge]:return [self.get_edge(i) for i in range(len(self._edges))]def change_edge(self, i: int, new_cap: int, new_flow: int) -> None:assert 0 <= i < len(self._edges)assert 0 <= new_flow <= new_cape = self._edges[i]e.cap = new_cap - new_flowassert e.rev is not Nonee.rev.cap = new_flowdef flow(self, s: int, t: int, flow_limit: Optional[int] = None) -> int:assert 0 <= s < self._nassert 0 <= t < self._nassert s != tif flow_limit is None:flow_limit = cast(int, sum(e.cap for e in self._g[s]))current_edge = [0] * self._nlevel = [0] * self._ndef fill(arr: List[int], value: int) -> None:for i in range(len(arr)):arr[i] = valuedef bfs() -> bool:fill(level, self._n)queue = []q_front = 0queue.append(s)level[s] = 0while q_front < len(queue):v = queue[q_front]q_front += 1next_level = level[v] + 1for e in self._g[v]:if e.cap == 0 or level[e.dst] <= next_level:continuelevel[e.dst] = next_levelif e.dst == t:return Truequeue.append(e.dst)return Falsedef dfs(lim: int) -> int:stack = []edge_stack: List[MFGraph._Edge] = []stack.append(t)while stack:v = stack[-1]if v == s:flow = min(lim, min(e.cap for e in edge_stack))for e in edge_stack:e.cap -= flowassert e.rev is not Nonee.rev.cap += flowreturn flownext_level = level[v] - 1while current_edge[v] < len(self._g[v]):e = self._g[v][current_edge[v]]re = cast(MFGraph._Edge, e.rev)if level[e.dst] != next_level or re.cap == 0:current_edge[v] += 1continuestack.append(e.dst)edge_stack.append(re)breakelse:stack.pop()if edge_stack:edge_stack.pop()level[v] = self._nreturn 0flow = 0while flow < flow_limit:if not bfs():breakfill(current_edge, 0)while flow < flow_limit:f = dfs(flow_limit - flow)flow += fif f == 0:breakreturn flowdef min_cut(self, s: int) -> List[bool]:visited = [False] * self._nstack = [s]visited[s] = Truewhile stack:v = stack.pop()for e in self._g[v]:if e.cap > 0 and not visited[e.dst]:visited[e.dst] = Truestack.append(e.dst)return visitedh,w=LI()aa=LLI(h)atoij=defaultdict(list)for i in range(h):for j in range(w):a=aa[i][j]if a==0:continueatoij[a].append((i,j))ans=0s=h+wt=h+w+1for a,ij in atoij.items():if len(ij)==1:ans+=1continuemf=MFGraph(h+w+2)ii,jj=set(),set()for i,j in ij:mf.add_edge(i,h+j,inf)ii.add(i)jj.add(j)for i in ii:mf.add_edge(s,i,1)for j in jj:mf.add_edge(h+j,t,1)ans+=mf.flow(s,t)print(ans)