[DP 组合] BZOJ5004 & Hihocoder1075. 开锁魔法 III

最新推荐文章于 2020-04-02 22:05:24 发布

LowestJN

最新推荐文章于 2020-04-02 22:05:24 发布

阅读量459

点赞数

分类专栏： DP 排列组合

本文链接：https://blog.csdn.net/Coldef/article/details/78636841

版权

DP 同时被 2 个专栏收录

84 篇文章 0 订阅

订阅专栏

排列组合

6 篇文章 0 订阅

订阅专栏

$i$ 向 $a_i$ 连边，会形成若干个环

那么把钥匙分配给这些环的概率就是答案

$f_{i,j}$ 表示前 $i$ 个环都打开了，还剩下 $j$ 把钥匙的概率

f i, j = f i - 1 , k \times ( s i z e i k - j ) ( n - s i z e i j ) ( n k )

$f_{i,j}={f_{i-1,k}\times {size_i\choose k-j}{n-size_i\choose j}\over{n\choose k}}$

这样DP是 $O(n^2)$ 的

不过BZOJ没有spj，被卡精度了…hihocoder上过了…

DP转移可以改一下

f i, j = f i - 1, k \times (s i z e i k - j)

$f_{i,j}=f_{i-1,k}\times {size_i\choose k-j}$

这样求出总方案数，最后除以 $n\choose k$ 就好了…

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>

using namespace std;

const int N=310;

int T,t,n,k,size[N],vis[N],a[N];
double f[N][N],C[N][N];

int dfs(int x){
  if(vis[x]) return 0;
  vis[x]=1; return dfs(a[x])+1;
}

int main(){
  scanf("%d",&T);
  C[1][1]=1; C[0][0]=1;
  for(int i=1;i<=300;i++) C[i][0]=1;
  for(int i=2;i<=300;i++)
    for(int j=1;j<=i;j++)
      C[i][j]=C[i-1][j]+C[i-1][j-1];
  while(T--){
    scanf("%d%d",&n,&k); int nn=n;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) vis[i]=size[i]=0;
    t=0;
    for(int i=1;i<=n;i++){
      if(vis[i]) continue;
      t++; size[t]=dfs(i);
    }
    for(int i=0;i<=n;i++)
      for(int j=0;j<=k;j++) f[i][j]=0;
    f[0][k]=1;
    for(int i=1;i<=t;i++){
      for(int j=1;j<=k;j++){
    if(f[i-1][j]<1e-7) continue;
    for(int s=1;s<=size[i] && s<=j;s++)
      f[i][j-s]+=f[i-1][j]*C[size[i]][s];
      }
      n-=size[i];
    }
    printf("%.9lf\n",f[t][0]/C[nn][k]);
  }
  return 0;
}