結果

問題 No.2263 Perms
ユーザー Haa
提出日時 2023-04-07 22:30:44
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 2,429 bytes
コンパイル時間 2,178 ms
コンパイル使用メモリ 179,672 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-02 19:57:21
合計ジャッジ時間 8,171 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 7 WA * 30 RE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
typedef pair<ll,ll> P;
typedef vector<ll> VI;
typedef vector<VI> VVI;
#define REP(i,n) for(int i=0;i<(n);i++)
#define ALL(v) v.begin(),v.end()
template<typename T> bool chmax(T& x, const T& y){return (x<y)?(x=y,true):false;};
template<typename T> bool chmin(T& x, const T& y){return (x>y)?(x=y,true):false;};
constexpr ll MOD=998244353;
constexpr ll INF=2e18;

int main(){
    int n, m; cin >> n >> m;
    VVI a(n,VI(n));
    REP(i,n)REP(j,n) cin >> a[i][j];
    REP(i,n){
        int sum=0;
        REP(j,n) sum+=a[i][j];
        if(sum!=m){
            cout << -1 << endl;
            return 0;
        }
        sum=0;
        REP(j,n) sum+=a[j][i];
        if(sum!=m){
            cout << -1 << endl;
            return 0;
        }
    }
    VVI ans(m,VI(n,-1));
    REP(i,m){
        if(i!=m-2){
            vector<P> p;
            REP(j,n)REP(k,n) p.push_back({a[j][k],j*n+k});
            sort(ALL(p));
            reverse(ALL(p));
            VI used(n,0);
            REP(l,n*n){
                int j=p[l].second/n, k=p[l].second%n;
                if(used[k])
                    continue;
                if(ans[i][j]!=-1)
                    continue;
                ans[i][j]=k;
                used[k]=1;
                a[j][k]--;
            }
        }
        else{
            int s=-1;
            REP(j,n)REP(k,n){
                if(a[j][k]==2){
                    a[j][k]--;
                    ans[i][j]=k;
                }
                else if(s==-1&&a[j][k]==1){
                    a[j][k]--;
                    ans[i][j]=k;
                    s=j;
                }
            }
            while(1){
                bool ok=0;
                REP(k,n){
                    if(a[s][k]==1){
                        REP(j,n){
                            if(j==s)
                                continue;
                            if(a[j][k]==1){
                                a[j][k]--;
                                ans[i][j]=k;
                                s=j;
                                ok=1;
                                break;
                            }
                        }
                        break;
                    }
                }
                if(!ok)
                    break;
            }
        }
    }
    REP(i,m)REP(j,n) cout << ans[i][j]+1 << " \n"[j==n-1];
}
0