結果

問題 No.74 貯金箱の退屈
ユーザー 37zigen
提出日時 2016-03-24 03:06:17
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 2,563 bytes
コンパイル時間 2,194 ms
コンパイル使用メモリ 80,036 KB
実行使用メモリ 55,300 KB
最終ジャッジ日時 2024-10-01 21:47:02
合計ジャッジ時間 9,437 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

package yukicoder074;
import java.util.ArrayList;
import java.util.Scanner;
public class Main {
	public static void main(String[] args){
		Scanner sc=new Scanner(System.in);
		int n=sc.nextInt();
		int[] d=new int[n];
		int[] w=new int[n];//k番目のコインの裏表（0の時裏、1の時表）
		double[][] Coe=new double[n][n+1];
		for(int i=0;i<n;i++){
			d[i]=sc.nextInt();
		}
		for(int i=0;i<n;i++){
			w[i]=sc.nextInt();
		}
		for(int i=0;i<n;i++){
			Coe[(i+d[i])%n][i]=1;
			int s=i-d[i];
			while(s<0){
				s+=n;
			}
			Coe[(s)%n][i]=1;
		}
		for(int i=0;i<n;i++){
			int a=0;
			if(w[i]==1)a=0;
			if(w[i]==0)a=1;
			Coe[i][n]=a;
		}
		//		for(int o=0;o<Coe.length;o++){
		//			for(int p=0;p<Coe[0].length;p++){
		//				System.out.print(Coe[o][p]+" ");
		//			}
		//			System.out.println();
		//		}
		if(gaussElimination(Coe)==true){
			System.out.println("Yes");
		}else{
			System.out.println("No");
		}
	}
	public static boolean gaussElimination(double[][] coe){
		for(int i=0;i<coe[0].length-1;i++){
			for(int j=i;j<coe.length;j++){
				if(coe[j][i]!=0){
					double[] tmp=coe[j];
					coe[j]=coe[i];
					coe[i]=tmp;
					break;
				}
			}
			double coe1=coe[i][i];
			if(coe1==0){
				continue;
			}
			for(int j=0;j<coe[0].length;j++){
				coe[i][j]=coe[i][j]/coe1;
			}
			for(int j=0;j<coe.length;j++){
				if(i==j)continue;
				double coe2=coe[j][i];
				for(int k=0;k<coe[0].length;k++){
					coe[j][k]=(coe[j][k]-coe2*coe[i][k])%2;
					while(coe[j][k]<0){
						coe[j][k]+=2;
					}
				}
			}
		}
//		for(int i=0;i<coe.length;i++){
//			for(int j=0;j<coe[0].length;j++){
//				System.out.print(coe[i][j]+" ");
//			}
//			System.out.println();
//		}
		ArrayList<Integer> unfixed=new ArrayList<>(); 
		double[] ans=new double[coe.length];
		for(int i=coe.length-1;i>=0;i--){
			double sum=0;
			for(int j=i+1;j<coe[0].length-1;j++){
				sum=(sum+coe[i][j]*ans[j])%2;
			}
			if(coe[i][i]==1){
				ans[i]=(-sum+coe[i][coe[0].length-1])%2;
				while(ans[i]<0)ans[i]+=2;
			}else if(coe[i][i]==0){
				if(-sum+coe[i][coe[0].length-1]!=0){
					System.err.println("対角成分がゼロかつ和が一致しない");
					System.err.println(i+"行");
					return false;
				}else {
					ans[i]=0;
					unfixed.add(i);
				}
			}else{
				System.err.println("ErrorAt90");
			}
		}
	
//		System.out.println("answer");
//		for(int i=0;i<coe.length;i++){
//			System.out.print(ans[i]+" ");
//		}
//		for(int i=0;i<coe.length;i++){
//			if((int)ans[i]!=ans[i]){
//				return false;
//			}
//		}
//		System.out.println();
		return true;
	}
}
0