ZOJ - 3605 Find the Marble 三维概率dp

题目链接：点击查看

题意：一共有 n 个杯子，有一个石头开始是放在第 s 号杯子里，然后交换这些杯子 m 次，但是你只能知道其中的k次，每次概率相同，最后输出概率最大的杯子编号，概率相同输出编号小的

题解：dp[i][j][k] 表示前i个选了j个第k个杯子的概率，那么前i个杯子第j个选和不选的概率是多少呢？选C(i-1,j-1) / C(i,j) = j / i;

注意要取eps  1e-8 

#include<bits/stdc++.h>
using namespace std;
#define eps 1e-8
int n,m,k,s;
double dp[55][55][55];
int a[55],b[55];
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d%d%d",&n,&m,&k,&s);
		for(int i=1;i<=m;i++) scanf("%d%d",&a[i],&b[i]);
		for(int i=0;i<=m;i++)
		{
			for(int j=0;j<=k;j++)
				for(int l=0;l<=n;l++)
					dp[i][j][l]=0;
			dp[i][0][s]=1;
		}
		for(int i=1;i<=m;i++)
		{
			for(int j=1;j<=min(i,k);j++)
			{
				for(int l=1;l<=n;l++)
				{
					if(a[i]==l) dp[i][j][l]=(double)j/i*dp[i-1][j-1][b[i]]+(double)(i-j)/i*dp[i-1][j][a[i]];
					else if(b[i]==l) dp[i][j][l]=(double)j/i*dp[i-1][j-1][a[i]]+(double)(i-j)/i*dp[i-1][j][b[i]];
					else dp[i][j][l]=(double)j/i*dp[i-1][j-1][l]+(double)(i-j)/i*dp[i-1][j][l];
				}
			}
		}
		int id=1;
		for(int i=2;i<=n;i++)
			if(dp[m][k][i]-dp[m][k][id]>eps)
				id=i;
		printf("%d\n",id);
		 
	}
	return 0;
}