POJ 3744 Scout YYF I 概率DP matrix 一开始考虑错误TLE了

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
//高精度注意-0.000000
//有序无序输入要注意啊
struct matrix
{
	int n,m;
	double mat[2][2];
	 void clear(void)
	{
		memset(mat,0,sizeof(mat));
	}
	void get_e(void)
	{
		int i=0;
		clear();
		for(i=0;i<n;i++)mat[i][i]=1;
	}
	matrix()
	{
		clear();
	}
	matrix(int x,int y):n(x),m(y)
	{
		clear();
	}
};
matrix operator*(const matrix &a,const matrix &b)
{
	int i,j,k;
	matrix ret(a.n,b.m);
	for(k=0;k<a.m;k++)
		for(i=0;i<a.n;i++)
			if(a.mat[i][k]!=0)
				for(j=0;j<b.m;j++)
					ret.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
	return ret;
}
matrix mypow(matrix a,int x)
{
	matrix ret(a.n,a.m);
		if(x<0)return ret;
	ret.get_e();
	while(x)
	{
		if(x&1)ret=ret*a;
		a=a*a;
		x>>=1;
	}
	return ret;
}
const int LMT=12;
int pos[LMT];
int main(void)
{
    int n;
    double p,ans;
    matrix tem(2,1),use(2,2);
    while(~scanf("%d%lf",&n,&p))
    {
        for(int i=1;i<=n;++i)
        scanf("%d",&pos[i]);
        sort(pos,pos+n+1);
        tem.clear();
        tem.mat[1][0]=1.0;
        use.mat[1][1]=p;
        use.mat[1][0]=1-p;
        use.mat[0][1]=1;
        for(int i=1;i<=n;i++)
        {
            ans=(mypow(use,pos[i]-pos[i-1]-2)*tem).mat[1][0];
            ans*=(1-p);
            tem.clear();
            tem.mat[1][0]=ans;
        }
        if(ans<0)ans=0;
        printf("%.7f\n",ans);
    }
    return 0;
}

一开始考虑错误,TLE
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值