結果

問題 No.241 出席番号(1)
ユーザー H3PO4
提出日時 2020-06-04 22:12:52
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 883 ms / 2,000 ms
コード長 506 bytes
コンパイル時間 117 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 56,652 KB
最終ジャッジ日時 2024-11-30 08:18:11
合計ジャッジ時間 29,661 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
外部呼び出し有り
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

from scipy.sparse import csr_matrix
from scipy.sparse.csgraph import maximum_bipartite_matching

N = int(input())
A = [int(input()) for _ in range(N)]

st = set(A)
if len(st) == 1 and st.pop() < N:
    print(-1)
    exit()

frm, to = [], []
for i, a in enumerate(A):
    for j in range(N):
        if j == a:
            continue
        frm.append(i)
        to.append(j)

matr = csr_matrix(([1] * len(frm), (frm, to)), shape=(N, N))
print(*maximum_bipartite_matching(matr, perm_type='column'), sep='\n')
0