题目链接:点击查看
题意:一共有 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;
}