結果

問題 No.813 ユキちゃんの冒険
ユーザー leaf_1415
提出日時 2019-04-16 02:27:04
言語 C++11
(gcc 4.8.5)
結果
WA   .
実行時間 -
コード長 1,097 Byte
コンパイル時間 298 ms
使用メモリ 9,432 KB
最終ジャッジ日時 2019-04-16 02:27:06

ソースコード

diff #
#include <iostream>
#include <math.h>
#include <stdio.h>

using namespace std;

int n;
double p, q;
double mat[1005][1005];

void swap(int i, int j)
{
	for(int k = 1; k <= n+1; k++){
		double t = mat[i][k];
		mat[i][k] = mat[j][k];
		mat[j][k] = t;
	}
}

int main(void)
{
	cin >> n >> p >> q;
	
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= n; j++){
			if(i==j)  mat[i][j] = 1;
			if(i+1==j) mat[i][j] = -q;
			if(i==j+1) mat[i][j] = -p;
		}
	}
	mat[1][n+1] = p;
	
	for(int i = 1; i <= n; i++){
		double max_val = 0; int max_j;
		for(int j = i; j <= n; j++){
			if(fabs(mat[j][i]) > max_val){
				max_val = fabs(mat[j][i]);
				max_j = j;
			}
		}
		if(max_val < 1e-9){
			cout << "Non-Regular" << endl;
			return 0;
		}
		swap(i, max_j);
		
		double div = mat[i][i];
		for(int j = i; j <= i+1; j++) mat[i][j] /= div;
		mat[i][n+1] /= div;
		
		for(int j = 1; j <= i+1; j++){
			double coe = mat[j][i];
			if(i == j) continue;
			for(int k = i; k <= i+1; k++){
				mat[j][k] -= coe * mat[i][k];
			}
			mat[j][n+1] -= coe * mat[i][n+1];
		}
	}
	
	printf("%.11f\n", mat[1][n+1]);
	return 0;
}
0