[dp+组合数学] hihocoder 1075 开锁魔法III

博客详细介绍了HihoCoder 1075题目的解题思路,重点在于如何找到环并将其转化为组合数学问题。通过动态规划方法,计算在有限次尝试中确保每个环内的盒子都被打开的方案数,最后将总方案数除以概率。文章提供了关键的转移方程和代码片段。
摘要由CSDN通过智能技术生成

题意:

中文题~

思路:

首先需要处理的就是把所有的环找出来

环的意思就是这个环里面只要有一个盒子被打开了,其他盒子就都被打开了。

比如样例。就是有两个环(1,2,5) 和 (3,4)并记录环内盒子的总数。

这样就转换成了这样的一个问题了。

给n堆东西,每堆有ai个,问现在取k次,保证每堆至少取1个的方案数。

然后总的方案数是C[n][k]。概率一除就好了。

那么这个方案数怎么求呢。

我们就要设dp[i][j]代表花j个钥匙开前i个环的方案数。

那么转移的话就是 dp[i][j]=Σ dp[i-1][j-k]*C[ai][k]  (1<=k<=ai)

代码:

#include"cstdio"
#include"cstring"
#include"cmath"
#include"cstdlib"
#include"algorithm"
#include"iostream"
#include"map"
#include"vector"
#include"queue"
#define ll long long
#define inf 9999999
using namespace std;
#define MAXN 100004
int loop[333],a[333],used[333];
double dp[333][333],c[333][333];
int main()
{
    int t;
    cin>>t;
    memset(c,0,sizeof(c));
    for(int i=0; i<=300; i++)
    {
        c[i][0]=c[i][i]=1.0;
        for(int j=1; j<i; j++)
            c[i][j]=c[i-1][j-1]+c[i-1][j];
    }
    while(t--)
    {
        int n,k;
        int cnt=0;
        scanf("%d%d",&n,&k);
        for(int i=1; i<=n; i++) scanf("%d",&a[i]);
        memset(used,0,sizeof(used));
        for(int i=1; i<=n; i++)
        {
            if(used[i]==1) continue;
            int j=i,sum=0;
            while(!used[j])
            {
                sum++;
                used[j]=1;
                j=a[j];
            }
            loop[cnt++]=sum;
        }
        memset(dp,0,sizeof(dp));
        dp[0][0]=1.0;
        for(int i=0;i<cnt;i++)
        {
            for(int j=0;j<=k;j++)
            {
                if(fabs(dp[i][j])<=0.000001) continue;
                for(int l=1;l<=loop[i]&&j+l<=k;l++)
                {
                    dp[i+1][j+l]+=dp[i][j]*c[loop[i]][l];
                }
            }
        }
        printf("%.4f\n",dp[cnt][k]/c[n][k]);
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值