CCF 202109-4（状态压缩 + 期望DP）-CSDN博客

本文链接：https://blog.csdn.net/qq_45604735/article/details/120590387

配套地址

在这里插入图片描述

将每一个卡牌是否选取用状压 $s$ 表示。

设 $d p (s, j)$ 表示到达状态 $(111 . . . . 11)$ 且当前硬币数为 j 个还需要选的次数。

那么

$\& 1 == 1：$
$\times p(z)$
$e l s e ：$
$\times p(z)$

最后
$d p (s, j) + = 1$

注意，当 $k))\ \ dp[i][j] = 0$
表示当前状态还需要0步即可到达最终状态 $(1111 . . . . 11111)$ 。

答案为 $d p (0, 0)$

#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include<bits/stdc++.h>
using namespace std;
const int N = 16;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
//#define int long long
typedef long long ll;
typedef pair<int,int> pii;

double dp[1<<N][N*N];
double p[N];

signed main(){
	IOS
	#ifdef ddgo
		freopen("C:\\Users\\asus\\Desktop\\ddgoin.txt","r",stdin);
    #endif
    int n,k; cin>>n>>k;
    for(int i=0;i<n;i++) cin>>p[i];
    dp[(1<<n)-1][0] = 0;
    for(int i=(1<<n)-2;i>=0;i--){
    	for(int j=k*n;j>=0;j--) {
	    	for(int z=0;z<n;z++) {
	    		if((i >> z) & 1) dp[i][j] += dp[i][j+1] * p[z];
	    		else dp[i][j] += dp[i | (1 << z)][j] * p[z];
			}
			dp[i][j] += 1;
			int cnt = n;
			for(int z=0;z<n;z++) if((i >> z) & 1) cnt --;
			if(j >= (cnt * k)) dp[i][j] = 0;
		}
	}
    cout<<fixed<<setprecision(10)<<dp[0][0]<<endl;
	return 0;
}