POJ 3744 分段矩阵乘法优化求概率

原创

于 2013-08-12 13:12:24 发布 · 2.3k 阅读

0 ·

CC 4.0 BY-SA版权

博客介绍了如何利用分段矩阵乘法优化解决POJ 3744问题，重点在于高效计算特定概率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意很简单的，很常见的矩阵做法

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int a[11];
void mult(double a[][2], double b[][2]) {  //矩阵乘法 a = a*b
	int i, j, k;
	double c[2][2] = {0.0};
	for(i = 0; i < 2; i++)
		for(k = 0; k < 2; k++) if(a[i][k] != 0.0) {
			for(j = 0; j < 2; j++)
				c[i][j] += a[i][k] * b[k][j];
		}
	for(i = 0; i < 2; i++)
		for(j = 0; j < 2; j++)
			a[i][j] = c[i][j];
}
int n;
double p;
double A[2][2],ans[2][2];
void solve(int n) {   // ans = A^n
	A[0][0] = p; A[0][1] = 1;
	A[1][0] = 1-p ;A[1][1] = 0;
	ans[0][0] = ans[1][1] = 1.0;
	ans[0][1] = ans[1][0] = 0;
	while(n) {
		if(n&1) mult(ans, A);
		mult(A, A);
		n >>= 1;
	}
}

int t;
int main() {
	int i, j;
	while( ~scanf("%d%lf", &n, &p)) {
		for(i = 0; i < n; i++) scanf("%d", &a[i]);
		sort(a, a+n);
		double ret = 1.0;
		sol