結果

問題 No.1078 I love Matrix Construction
ユーザー mugen_1337mugen_1337
提出日時 2020-06-12 23:09:47
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,985 bytes
コンパイル時間 2,326 ms
コンパイル使用メモリ 211,084 KB
実行使用メモリ 23,516 KB
最終ジャッジ日時 2024-06-24 05:52:25
合計ジャッジ時間 6,704 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 WA -
testcase_02 AC 8 ms
6,400 KB
testcase_03 AC 24 ms
10,992 KB
testcase_04 AC 39 ms
14,064 KB
testcase_05 WA -
testcase_06 AC 8 ms
6,940 KB
testcase_07 AC 4 ms
6,944 KB
testcase_08 AC 23 ms
12,040 KB
testcase_09 WA -
testcase_10 AC 56 ms
23,516 KB
testcase_11 AC 29 ms
14,660 KB
testcase_12 AC 45 ms
20,132 KB
testcase_13 AC 66 ms
22,216 KB
testcase_14 AC 39 ms
16,264 KB
testcase_15 AC 62 ms
21,340 KB
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 7 ms
6,940 KB
testcase_19 AC 14 ms
8,192 KB
testcase_20 AC 12 ms
8,064 KB
testcase_21 AC 2 ms
6,944 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;
    };
    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+n*n;
            if(check[ot]){
                check[now]=false;
                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){
            // rep(I,n)rep(J,n) cout<<I<<","<<J<<" : "<<check[get(I,J)]<<" "<<check[get(I,J,1)]<<endl;
            // cout<<i<<", "<<j<<endl;
            if(check[get(i,j)] or check[get(i,j,1)]) continue;
            check[get(i,j)]=true;
            auto res=dfs(get(i,j));
            if(res) continue;
            check[get(i,j)]=false;
            check[get(i,j,1)]=true;
            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