結果

問題 No.1078 I love Matrix Construction
ユーザー mugen_1337mugen_1337
提出日時 2020-06-12 22:50:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,724 bytes
コンパイル時間 2,321 ms
コンパイル使用メモリ 209,656 KB
実行使用メモリ 27,004 KB
最終ジャッジ日時 2023-09-06 10:58:10
合計ジャッジ時間 6,282 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 7 ms
6,732 KB
testcase_03 AC 17 ms
12,328 KB
testcase_04 AC 22 ms
15,952 KB
testcase_05 WA -
testcase_06 AC 8 ms
6,472 KB
testcase_07 AC 3 ms
4,468 KB
testcase_08 AC 19 ms
13,524 KB
testcase_09 WA -
testcase_10 AC 49 ms
27,004 KB
testcase_11 AC 27 ms
16,376 KB
testcase_12 AC 41 ms
22,848 KB
testcase_13 AC 55 ms
25,720 KB
testcase_14 AC 38 ms
18,492 KB
testcase_15 AC 45 ms
24,236 KB
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 6 ms
5,692 KB
testcase_19 AC 11 ms
8,980 KB
testcase_20 AC 12 ms
8,984 KB
testcase_21 AC 2 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define ALL(x) x.begin(),x.end()
#define rep(i,n) for(int i=0;i<(n);i++)
#define debug(v) cout<<#v<<":";for(auto x:v){cout<<x<<' ';}cout<<endl;
#define INF 1000000000
#define mod 1000000007
using ll=long long;
const ll LINF=1001002003004005006ll;
int dx[]={1,0,-1,0};
int dy[]={0,1,0,-1};
// ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(b<a){a=b;return true;}return false;}

// 可能なら出ていくのがないところからテキトーに構築して終わる．
struct UnionFind{
    vector<int> par;
    vector<int> siz;
    int c;//親の個数(グループ数)
    void init(int n){
        c=n;
        for(int i=0;i<n;i++){
            par.push_back(i);
            siz.push_back(1);
        }
    }
    UnionFind(int n){
        init(n);
    }
    int root(int x){
        //根を返す
        return(par[x]==x?x:(par[x]=root(par[x])));
    }
    bool sameParent(int x,int y){
        return(root(x)==root(y));
    }
    void unite(int x, int y){
        x=root(x);y=root(y);
        if(x==y)    return;
        if(siz[x]<siz[y]) swap(x,y);
        siz[x] += siz[y];par[y]=x;c--;
    }
    int size(int x){
        return siz[root(x)];
    }
};

signed main(){
    cin.tie(0);
    ios::sync_with_stdio(0);
    
    int n;cin>>n;
    auto get=[&](int i,int j,bool on=false){
        int ret=i*n+j;
        if(on) ret+=n*n;
        return ret;
    };
    UnionFind uf(2*n*n);
    vector<vector<int>> g(2*n*n);
    vector<int> s(n),t(n),u(n);
    rep(i,n) cin>>s[i];
    rep(i,n) cin>>t[i];
    rep(i,n) cin>>u[i];
    rep(i,n) s[i]--,t[i]--;
    rep(i,n){
        if(u[i]==0){
            rep(j,n){
                // s=1->t=0
                // t=1->s=0
                g[get(s[i],j,1)].push_back(get(j,t[i]));
                g[get(j,t[i],1)].push_back(get(s[i],j));
            }
        }if(u[i]==1){
            rep(j,n){
                // s=1->t=1
                // t=0->s=0
                g[get(s[i],j,1)].push_back(get(j,t[i],1));
                g[get(j,t[i])].push_back(get(s[i],j));
            }
        }if(u[i]==2){
            rep(j,n){
                // s=0->t=0
                // t=1->s=1
                g[get(s[i],j)].push_back(get(j,t[i]));
                g[get(j,t[i],1)].push_back(get(s[i],j,1));
            }

        }if(u[i]==3){
            rep(j,n){
                // s=0->t=1
                // t=0->s=1
                g[get(s[i],j)].push_back(get(j,t[i],1));
                g[get(j,t[i])].push_back(get(s[i],j,1));
            }
        }
    }
    
    vector<bool> check(2*n*n,false);

    function<bool(int)> dfs=[&](int now){
        for(auto to:g[now]){
            int ot;
            if(to>=n*n) ot=to-n*n;
            else ot=to;
            if(check[ot]){
                return false;
            }
            if(check[to]) continue;
            check[to]=true;
            if(!dfs(to)){
                check[now]=false;
                return false;
            }
        }
        return true;
    };

    rep(i,n){
        rep(j,n){
            if(check[get(i,j)] or check[get(i,j,1)]) continue;
            auto res=dfs(get(i,j));
            if(res) continue;
            res|=dfs(get(i,j,1));
            if(!res){
                cout<<-1<<endl;
                return 0;
            }
        }
    }
    rep(i,n){
        rep(j,n){
            if(!check[get(i,j)] and !check[get(i,j,1)]) cout<<0<<(j+1==n?"\n":" ");
            if(check[get(i,j)])     cout<<0<<(j+1==n?"\n":" ");
            if(check[get(i,j,1)])   cout<<1<<(j+1==n?"\n":" ");
        }
    }
    return 0;
}
0