#3579 (C++11) No.75 回数の期待値の問題

• 問題: No.75 回数の期待値の問題
• ユーザー: ciel
• 提出日時: 2014-11-24 23:23:00
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 6 ms / 5,000 ms
• コード長: 1,446 bytes
• コンパイル時間: 486 ms
• コンパイル使用メモリ: 49,024 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-06-10 22:06:02
• 合計ジャッジ時間: 1,139 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	1 ms; 5,376 KB
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	1 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	3 ms; 5,376 KB
testcase_18	AC	5 ms; 5,376 KB
testcase_19	AC	6 ms; 5,376 KB

コンパイルメッセージ

main.cpp: In function ‘int main()’:
main.cpp:65:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   65 |         scanf("%d",&N);
      |         ~~~~~^~~~~~~~~

ソースコード

#include <vector>
#include <cstdio>
#include <cmath>
using namespace std;

//derived from checkio expected-dice
//unfortunately TLE using Python, rewriting in C++11.

bool gauss(vector<vector<double> > &a){
	if(a.empty())return false;
	int n=a.size();
	for(int i=0;i<n;i++){
		if(!a[i][i]){
			int j=i+1;
			for(;j<n;j++){
				if(a[j][i]){
					for(int k=i;k<=n;k++)a[i][k]+=a[j][k];
					break;
				}
			}
			if(j<n)return false;
		}
		for(int j=0;j<n;j++)if(i!=j){
			double r=a[j][i]/a[i][i];
			for(int k=i;k<=n;k++)a[j][k]=a[j][k]-a[i][k]*r;
		}
	}
	for(int i=0;i<n;i++){
		double x=a[i][i];
		for(int j=0;j<a[i].size();j++)a[i][j]/=x;
	}
	return true;
}

double expected(int n, int s, int t){//vector<int> b){
	//perform checkio probably-dice
	vector<double> a(s*(n+1)+1);
	for(int i=1;i<=s;i++)a[i+s]=pow(1.0/s,n);
	for(int e=0;e<n-1;e++)for(int i=s*n;i>=0;i--){
		double sum=0;
		for(int j=i;j<i+s;j++)sum+=a[j];
		a[i+s]=sum;
	}
	int l=t+1; //b.size();
	vector<vector<double> > A(l);
	for(int i=0;i<l;i++){
		A[i].resize(l+1);
		A[i][i]=1;
		if(i!=t)A[i][l]=1;
	}
	for(int start=0;start<l;start++)if(start!=t){
		for(int roll=n;roll<=s*n;roll++){
			int end=roll+start;
			//end = (end + b[end%l])%l;
			if(end>=l)end=0;
			A[start][end]-=a[roll+s];
		}
	}
	gauss(A);
	return A[0][A[0].size()-1];
}

int main(){
	int N;
	scanf("%d",&N);
	printf("%f\n",expected(1,6,N));
}