5004: 开锁魔法II 概率DP

最新推荐文章于 2019-03-13 17:48:55 发布

200815147

最新推荐文章于 2019-03-13 17:48:55 发布

阅读量254

点赞数

分类专栏： DP 概率与期望

本文链接：https://blog.csdn.net/baidu_36797646/article/details/78811878

版权

DP 同时被 2 个专栏收录

86 篇文章 0 订阅

订阅专栏

概率与期望

10 篇文章 0 订阅

订阅专栏

题解：

转化一下题意，若 $i$ 向 $a_i$ 连一条边，那么最后就会变成若干个环（因为每个点的入度出度都为1），那么选中环中的一个点就相当于选中了整个环，问选 $k$ 次选中所有环的概率。求概率有多种方式，本题采用合法方案数/总方案数，那么 $f[i][j]$ 表示前 $i$ 个环用了 $j$ 次选完的方案数，转移的话枚举第 $i$ 个环被选中了多少次即可，由于方案数巨大，组合数和 $f$ 都要用double类型。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=310;
const int inf=2147483647;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int n,k,a[Maxn];
bool mark[Maxn];
int b[Maxn],c=0;
double C[Maxn][Maxn],f[Maxn][Maxn];//f[i][j]前i个环用j次解决的方案数 
void dfs(int x)
{
    mark[x]=true;
    b[c]++;
    if(!mark[a[x]])dfs(a[x]);
}
int main()
{
    memset(C,0,sizeof(C));
    C[0][0]=1.0;
    for(int i=1;i<=300;i++)
    {
        C[i][0]=1.0;
        for(int j=1;j<=i;j++)
        C[i][j]=C[i-1][j-1]+C[i-1][j];
    }
    int T=read();
    while(T--)
    {
        memset(b,0,sizeof(b));
        memset(f,0,sizeof(f));
        memset(mark,false,sizeof(mark));
        n=read(),k=read();c=0;
        for(int i=1;i<=n;i++)a[i]=read();
        for(int i=1;i<=n;i++)if(!mark[i])c++,dfs(i);
        f[0][0]=1;
        for(int i=1;i<=c;i++)
        {
            for(int j=1;j<=k;j++)
            {
                for(int l=1;l<=b[i];l++)
                if(l>j)break;
                else f[i][j]+=f[i-1][j-l]*C[b[i]][l];
            }
        }
        printf("%.9lf\n",f[c][k]/C[n][k]);
    }
}