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

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

#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;
solve(a[0]-1);
ret *= (1-ans[0][0]);
for(i = 1; i < n; i++) {
solve(a[i]-a[i-1]-1);
ret *= (1-ans[0][0]);
}
printf("%.7f\n", ret);
}
return 0;
}