1. 程式人生 > >【概率dp】【滾動數組】CDOJ1652 都市大飆車

【概率dp】【滾動數組】CDOJ1652 都市大飆車

ima 空間 pac names puts 都市 for 1.0 images

技術分享

轉移方程很顯然。

因為是多段圖模型,所以可以滾動數組優化一維空間。

#include<cstdio>
#include<cstring>
using namespace std;
int m,K,n,p;
bool zaw[1010][30010];
double f[2][30010];
int main(){
	int x,y;
	bool flag=1;
	scanf("%d%d%d%d",&m,&K,&n,&p);
	for(int i=1;i<=K;++i){
		scanf("%d%d",&x,&y);
		if(y<=n){
			zaw[y][x]=1;
			flag=0;
		}
	}
	if(m==1){
		if(!flag){
			puts("0.000000");
		}
		else{
			puts("1.000000");
		}
		return 0;
	}
	bool cur=0;
	f[cur][p]=1;
	for(int i=1;i<=n;++i){
		cur^=1;
		memset(f[cur],0,sizeof(f[cur]));
		for(int k=1;k<=2;++k){
			if(!zaw[i-1][1]){
				f[cur][k]+=f[cur^1][1]/2.0;
			}
		}
		for(int k=m-1;k<=m;++k){
			if(!zaw[i-1][m]){
				f[cur][k]+=f[cur^1][m]/2.0;
			}
		}
		for(int j=2;j<m;++j){
			for(int k=j-1;k<=j+1;++k){
				if(!zaw[i-1][j]){
					f[cur][k]+=f[cur^1][j]/3.0;
				}
			}
		}
	}
	double ans=0;
	for(int i=1;i<=m;++i){
		if(!zaw[n][i]){
			ans+=f[cur][i];
		}
	}
	printf("%.6lf\n",ans);
	return 0;
}

【概率dp】【滾動數組】CDOJ1652 都市大飆車