結果

問題 No.74 貯金箱の退屈
ユーザー koyumeishikoyumeishi
提出日時 2014-11-24 23:29:22
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 3,032 bytes
コンパイル時間 769 ms
コンパイル使用メモリ 83,280 KB
実行使用メモリ 6,940 KB
最終ジャッジ日時 2024-06-10 22:06:10
合計ジャッジ時間 1,748 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 WA -
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 1 ms
5,376 KB
testcase_18 AC 1 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 1 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 2 ms
5,376 KB
testcase_30 AC 3 ms
5,376 KB
testcase_31 AC 2 ms
5,376 KB
testcase_32 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cstdio>
#include <sstream>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
#include <cmath>

using namespace std;

void print(const vector<vector<int> > &v){
	int n = v.size();
	int m = v[0].size();
	for(int i=0; i<n; i++){
		for(int j=0; j<m; j++){
			cerr << v[i][j] << (j==(n-1)? " | " : " ");
		}
		cerr << endl ;
	}
	cerr << endl;
	
}

//A[n*n]*b[n*1] = x[n*1]
//A^(-1)[n*n]*x[n*1] = b[n*1]
bool gaussian_elimination(vector<vector<int> > &A, vector<vector<int> > &x){
	bool valid = true;
	int rank=0;
	int n = A.size();
	vector< vector<int> > R = A;
	for(int i=0; i<n; i++){
		R[i].insert(R[i].end(), x[i].begin(), x[i].end());
	}
	int m = R[0].size();

	//print(R);

	//foward
	//row
	for(int i=0; i<n; i++){

		//print(R);
		
		//pivot
		int pivot_row = i;
		int pivot_val = abs(R[i][i]);
		//choose pivot row
		for(int j=i+1; j<n; j++){
			if(pivot_val < abs(R[j][i])){
				pivot_row = j;
				pivot_val = abs(R[j][i]);
			}
		}
		if(pivot_row != i){
			swap(R[i], R[pivot_row]);
		}

		if(pivot_val == 0){
			
			continue;
		}

		rank++;

		{
			int divisor = 1/R[i][i];
			//A[i][j] == 0, such that (0 <= j < i)
			for(int j=i; j<m; j++){
				R[i][j] *= divisor;
				R[i][j] %= 2;
			}
		}
		
		for(int j=i+1; j<n; j++){
			int divisor = R[j][i];	// R[j][i]/R[i][i], but R[i][i] is 1.0
			for(int k=i; k<m; k++){
				R[j][k] -= R[i][k] * divisor;
				R[j][k] %= 2;
			}
		}
	}

	//print(R);

	//backward
	//row
	for(int i=n-1; i>=0; i--){
		for(int j=i-1; j>=0; j--){
			int c = R[j][i];
			for(int k=i; k<m; k++){
				R[j][k] -= R[i][k] * c;
			}
		}
	}

	for(int i=0; i<n; i++){
		bool row_valid = false;
		for(int j=i; j<m; j++){
			if(abs(R[i][j]) <= 1e-9){
				R[i][j] = 0.0;
			}else if(j<n){
				row_valid = true;
				break;
			}
		}
		if(row_valid == false && R[i][n] != 0){
			valid = false;
			break;
		}
	}

	//print(R);
	//cerr << "rank is " << rank << endl;
	//cerr << (valid?"valid":"invalid, no solutions") << endl;

	return valid;
}



int main(){
	int N;
	cin >> N;
	vector<int> D(N);
	for(int i=0; i<N; i++){
		cin >> D[i];
		D[i] %= N;
	}

	//W[i][0] := i番目のコインをひっくり返さねばならないとき1、それ以外は0
	vector< vector<int> > W(N, vector<int>(1));
	for(int i=0; i<N; i++){
		cin >> W[i][0];
		//今の状態が表(入力:1)なら裏返す必要はないので0、裏(入力:0)なら裏返す必要があるので1
		W[i][0] = !W[i][0];
	}

	//C[i][j] := j枚目のコインでi番目のコインがひっくり返る場合1、それ以外は0
	vector< vector<int> > C(N, vector<int>(N, 0));
	for(int i=0; i<N; i++){
		C[ (i+D[i])%N ][i] |= 1;
		C[ (i-D[i]+N)%N ][i] |= 1;
	}

	//ガウスの消去法でCx = Wを解いて解が存在するなら"Yes"、そうでなければ"No"
	cout << (gaussian_elimination(C,W)?"Yes":"No") << endl;
	return 0;
}
0