#496208 (C++17) No.1078 I love Matrix Construction

提出ソース

結果

問題	No.1078 I love Matrix Construction
ユーザー	kmjp
提出日時	2020-06-12 23:29:56
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	373 ms / 2,000 ms
コード長	2,402 bytes
コンパイル時間	2,565 ms
コンパイル使用メモリ	214,500 KB
実行使用メモリ	77,032 KB
最終ジャッジ日時	2024-06-24 06:03:14
合計ジャッジ時間	8,306 ms
ジャッジサーバーID （参考情報）	judge4 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	11 ms 27,972 KB
testcase_01	AC	11 ms 28,104 KB
testcase_02	AC	38 ms 37,092 KB
testcase_03	AC	110 ms 46,300 KB
testcase_04	AC	169 ms 53,280 KB
testcase_05	AC	133 ms 49,812 KB
testcase_06	AC	41 ms 34,668 KB
testcase_07	AC	20 ms 30,408 KB
testcase_08	AC	146 ms 48,944 KB
testcase_09	AC	15 ms 31,052 KB
testcase_10	AC	373 ms 77,032 KB
testcase_11	AC	172 ms 55,172 KB
testcase_12	AC	286 ms 68,524 KB
testcase_13	AC	332 ms 73,668 KB
testcase_14	AC	217 ms 59,476 KB
testcase_15	AC	319 ms 71,316 KB
testcase_16	AC	18 ms 29,896 KB
testcase_17	AC	11 ms 28,236 KB
testcase_18	AC	29 ms 32,972 KB
testcase_19	AC	63 ms 39,648 KB
testcase_20	AC	64 ms 40,984 KB
testcase_21	AC	12 ms 28,360 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;
typedef signed long long ll;

#undef _P
#define _P(...) (void)printf(__VA_ARGS__)
#define FOR(x,to) for(x=0;x<(to);x++)
#define FORR(x,arr) for(auto& x:arr)
#define ITR(x,c) for(__typeof(c.begin()) x=c.begin();x!=c.end();x++)
#define ALL(a) (a.begin()),(a.end())
#define ZERO(a) memset(a,0,sizeof(a))
#define MINUS(a) memset(a,0xff,sizeof(a))
//-------------------------------------------------------

class SCC {
public:
	static const int MV = 500*500*2;
	vector<vector<int> > SC; int NV,GR[MV];
private:
	vector<int> E[MV], RE[MV], NUM; int vis[MV];
public:
	void init(int NV) { this->NV=NV; for(int i=0;i<NV;i++) { E[i].clear(); RE[i].clear();}}
	void add_edge(int x,int y) { E[x].push_back(y); RE[y].push_back(x); }
	void dfs(int cu) { vis[cu]=1; for(int i=0;i<E[cu].size();i++) if(!vis[E[cu][i]]) dfs(E[cu][i]); NUM.push_back(cu); }
	void revdfs(int cu, int ind) { int i; vis[cu]=1; GR[cu]=ind; SC[ind].push_back(cu);
		FOR(i,RE[cu].size()) if(!vis[RE[cu][i]]) revdfs(RE[cu][i],ind);}
	void scc() {
		int c=0,i; SC.clear(); SC.resize(NV); NUM.clear();
		assert(NV);
		FOR(i,NV) vis[i]=0; FOR(i,NV) if(!vis[i]) dfs(i); FOR(i,NV) vis[i]=0;
		for(int i=NUM.size()-1;i>=0;i--) if(!vis[NUM[i]]){
			SC[c].clear(); revdfs(NUM[i],c); sort(SC[c].begin(),SC[c].end()); c++;
		}
		SC.resize(c);
	}
};

class TwoSat {
	int NV;
	SCC sc;
public:
	vector<int> val;
	void init(int NV) { this->NV=NV*2; sc.init(NV*2); val.resize(NV);}
	void add_edge(int x,int y) { // k+0:normal k+NV:inverse
		sc.add_edge((x+NV/2)%NV,y);
		sc.add_edge((y+NV/2)%NV,x);
	}
	bool sat() { // empty:false 
		sc.scc();
		for(int i=0;i<NV/2;i++) if(sc.GR[i]==sc.GR[i+NV/2]) return false;
		for(int i=0;i<NV/2;i++) val[i]=sc.GR[i]>sc.GR[i+NV/2];
		return true;
	}
};

TwoSat ts;
int N;
int S[505],T[505],U[505];

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>N;
	ts.init(N*N);
	FOR(i,N) cin>>S[i];
	FOR(i,N) cin>>T[i];
	FOR(i,N) {
		cin>>U[i];
		S[i]--;
		T[i]--;
		FOR(j,N) ts.add_edge(S[i]*N+j+(U[i]&1)*N*N,j*N+T[i]+(U[i]/2)*N*N);
	}
	
	if(ts.sat()) {
		FOR(y,N) {
			FOR(x,N) cout<<ts.val[y*N+x]<<" ";
			cout<<endl;
		}
	}
	else {
		cout<<-1<<endl;
	}
	
}


int main(int argc,char** argv){
	string s;int i;
	if(argc==1) ios::sync_with_stdio(false), cin.tie(0);
	FOR(i,argc-1) s+=argv[i+1],s+='\n'; FOR(i,s.size()) ungetc(s[s.size()-1-i],stdin);
	cout.tie(0); solve(); return 0;
}

yukicoder

結果

テストケース

ソースコード