結果

問題 No.1479 Matrix Eraser
ユーザー ygd.
提出日時 2021-04-17 14:56:11
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 823 bytes
コンパイル時間 87 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-07-03 22:37:13
合計ジャッジ時間 3,190 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample RE * 2
other RE * 39
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict
import networkx as nx
H,W = map(int,input().split())
A = []
dic = defaultdict(list) 
for i in range(H):
    a = list(map(int,input().split()))
    for j in range(W):
        num = a[j]
        if num != 0:
            dic[num].append((i,j))
    A.append(a)
print(A)
print(dic)
key_list = list(dic.keys())
#頂点は0~H+W-1
S = H+W
T = H+W + 1
ans = 0
for num in key_list:
    G1 = nx.Graph()
    G1 = nx.DiGraph(G1)
    SX = set([])
    SY = set([])
    for x,y in dic[num]:
        SX.add(x)
        SY.add(y)
        G1.add_edge(x,y+H,capacity=1)
    print(SX,SY)
    for x in SX:
        G1.add_edge(S,x,capacity=1)
    for y in SY:
        G1.add_edge(y+H,T,capacity=1)
    flow_value, flows = nx.maximum_flow(G1,S,T)
    #print(num,flow_value,flows)
    ans += flow_value
print(ans)
0