版权声明:转载请注明出处 https://blog.csdn.net/FSAHFGSADHSAKNDAS/article/details/82148979
链接
https://icpcarchive.ecs.baylor.edu/external/81/8138.pdf
题解
假设还有
种数字需要出现两次,
种数字需要出现一次
如果我用
表示目前还有
种数字需要出现两次,还有
种数字需要出现一次,到达目标状态的期望步数,那么写出转移就会发现,这个
的边界和
有关,那就必须每次
,复杂度
,承受不了
那我就重新设计状态,
表示
的时候,到达目标状态的期望步数,这样写出方程:
那个
可以最后再乘,预处理的时候直接写
就行,因为这个
其实就是对
的系数不断地累加或者乘以一个常数,所以我可以先计算系数最后乘以
,效果是一样的
代码
#include <bits/stdc++.h>
#define maxn 3010
#define maxk 100010
#define eps 1e-8
#define cl(x) memset(x,0,sizeof(x))
using namespace std;
double f[maxn][maxn];
int N, K, a[maxk], cnt[maxn];
void init()
{
int i, j;
for(i=0;i<=3000;i++)
{
for(j=0;j<=i;j++)
{
if(i==0 and j==0)continue;
f[i][j]=1.0/i;
if(j!=0)f[i][j]+=1.0*j/i*f[i][j-1];
if(i!=0)f[i][j]+=1.0*(i-j)/i*f[i-1][j];
}
}
}
void solve()
{
int i, cnt1=0, cnt2, cnt3;
cnt2=N;
for(i=1;i<=K;i++)
{
scanf("%d",a+i);
cnt[a[i]]++;
if(cnt[a[i]]==1)cnt2--, cnt1++;
if(cnt[a[i]]==2)cnt1--;
}
for(i=1;i<=K;i++)cnt[a[i]]--;
printf("%.10lf\n",N*f[cnt2+cnt1][cnt2]);
}
int main()
{
int T;
scanf("%d",&T);
init();
while(T--)scanf("%d%d",&N,&K), solve();
return 0;
}