結果

問題 No.2263 Perms
コンテスト
ユーザー とりゐ
提出日時 2023-04-04 17:04:00
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 7 ms / 2,000 ms
コード長 925 bytes
コンパイル時間 2,110 ms
コンパイル使用メモリ 210,584 KB
最終ジャッジ日時 2025-02-11 22:54:38
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/maxflow>
using namespace std;
using namespace atcoder;

int main(){
  int n,m;
  cin>>n>>m;
  int a[n][n];
  vector<int>low_sum(n,0),col_sum(n,0);
  for(int i=0;i<n;i++){
    for(int j=0;j<n;j++){
      cin>>a[i][j];
      low_sum[i]+=a[i][j];
      col_sum[j]+=a[i][j];
    }
  }
  for(int i=0;i<n;i++){
    if(low_sum[i]!=m||col_sum[i]!=m){
      cout<<"-1"<<endl;
      return 0;
    }
  }
  for(int x=0;x<m;x++){
    mf_graph<int>g(2*n+2);
    int S=2*n,T=2*n+1;
    for(int i=0;i<n;i++){
      g.add_edge(S,i,1);
      g.add_edge(i+n,T,1);
      for(int j=0;j<n;j++){
        if(a[i][j]>0)g.add_edge(i,n+j,1);
      }
    }
    g.flow(S,T,n);
    int P[n];
    for(auto e:g.edges()){
      if(0<=e.from&&e.from<n&&n<=e.to&&e.to<2*n&&e.flow==1){
        P[e.from]=e.to-n+1;
        a[e.from][e.to-n]-=1;
      }
    }
    for(int i=0;i<n;i++)cout<<P[i]<<(i==n-1?"\n":" ");
  }
}
0