老实说,经过一番WA哈哈后,独立A了一道题,这种写题解的冲动真是忍不住,好久没独立A了,呜呜~~~不废话了,贴码————
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <string>
#include <vector>
#include <set>
#include <map>
using namespace std;
typedef long long ll;
const int N = 5;//→_→其实好像可以不用5这么多
int a[20];//存放不能走的坐标
map<int, int>vis;//判断是否能走,能走就是0,不能就是1
//很常见的矩阵快速幂重载模板
struct node {
double jz[N][N];
node operator * (const node& m)const
{
node ans;
for(int i = 0; i < N; i++) {
for(int j = 0; j < N; j++) {
ans.jz[i][j] = 0;
for(int k = 0; k < N; k++) {
ans.jz[i][j] += jz[i][k] * m.jz[k][j];
}
}
}
return ans;
}
};
//第一行x[],其余全是0的矩阵*pow(c,k);快速幂
void Pow(double x[], node c, int k)
{
node ans;
memset(ans.jz, 0, sizeof(ans.jz));
for(int i = 0; i < N; i++)
ans.jz[0][i] = x[i];
while(k) {
if(k % 2)
ans = ans * c;
c = c * c;
k /= 2;
}
for(int i = 0; i < N; i++)
x[i] = ans.jz[0][i];
}
int main()
{
int n;
double p;
node c;
memset(c.jz, 0, sizeof(c.jz));//初始化常量矩阵
for(int i = 1; i < N; i++)
c.jz[i][i - 1] = 1;//左移一行,如:{a,b,c,d,e}->{b,c,d,e,f}
while(cin >> n >> p) {
a[0] = -20;
vis.clear();
c.jz[3][4] = 1 - p;
c.jz[4][4] = p;
//最后一个数,即f = d*(1-p) + e*p;
for(int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
vis[a[i]] = 1;
}
sort(a + 1, a + n + 1);
double f[N] = {0, 0, 0, 0, 1};//末尾1代表安全走到1位置概率为1(不管1能不能走,只是走到),之前以为是0→_→被坑了好久
a[n + 1] = a[n] + 1;//最后答案当然是走到a[n+1]
Pow(f, c, a[1] + 1);//→_→这个可以自己调,反正我是调到最终f[2]处对应a[1]
for(int i = 1; i <= n; i++) {
for(int j = 3; j < N; j++)
f[j] = !vis[a[i] - 1 + (j - 2)] * f[j - 1] * p + !vis[a[i] - 2 + (j - 2)] * f[j - 2] * (1 - p);
/*
vis[a[i]-1+(j-2)]判断当前能不能走,能走返回0,随意要取逻辑反,表示要计算,否则=0舍去
因为原来的f[3],f[4]都是在f[2]能走的情况下取的,很明显,此刻f[2]不能取(a[i]对应f[2]),所以更新f[3],f[4]
*/
Pow(f, c, a[i + 1] - a[i]);//调到f[2]处对应a[i+1],所以最终f[2]就是答案
}
printf("%.7f\n", f[2]);
}
return 0;
}