“蔚来杯“2022牛客暑期多校训练营1 I Chiitoitsu(概率dp)

"蔚来杯"2022牛客暑期多校训练营1 I Chiitoitsu(概率dp)

题意： 开局给你 $13$ 张牌，相同牌最多两张。加上手上的 $13$ 张牌一共是 $136$ 张，即一共有 $34$ 种类型，每种类型牌有 $4$ 张。求凑成 $7$ 对不同类型的牌的期望是多少

思路： 首先每一个测试给定的字符串是 唯一的 ，且相同牌数最多两张。那么显然只要开局牌数为 $2$ 的牌出现的次数相同，不管是什么牌，那么最终的期望都是相同的。我们考虑用 概率dp 来解决本问题。假设 $dp[i][j]$ ，$i$ 是指手上牌数为 $2$ 的牌的数量， $j$ 是指桌面上还有多少牌可以选择，即初始还有可以选择的牌数为 $136-13=123$ ，那么假定所求字符串中牌数为 $2$ 的牌的数量为 $i$，那么我们最终所求的期望是 $dp[i][123]$ 。所以我们需要做的就是预处理 $dp$ 数组。一共是 $O(7\ast 123)$ 的复杂度，然后每次统计出牌数为 $2$ 的牌的数量，然后 $O(1)$ 查询即可。接下去我们推导 dp转移方程 最好的策略是手上有单牌，我们将单排打出，然后拿到一张牌可以和手上另一单牌形成对子。即 $dp[i][j]+=dp[i+1][j-1]$ $\ast$ $(拿到对子的概率)$，另一情况是从桌面拿了一张牌但无法形成对子，将该牌打出即 $dp[i][j]+=dp[i][j-1]$ $\ast$ $(拿不到对子的概率)$。 所以转移方程为 $dp[i][j]=d[i+1][j-1]$ $\ast$ $(拿到对子的概率)$ +$dp[i][j-1]$ $\ast$ $(拿不到对子的概率)$。

代码：

/*
 * @author:  Snow
 * @LastEditTime: 2022-07-19 16:39:26
 * @Description:  Algorithm Contest
 */
#include<bits/stdc++.h>
using namespace std;
#define endl "\n"
#define int long long 
#pragma GCC optimize(3)
typedef pair<int,int>PII;
#define pb emplace_back
int dp[10][150];
const int mod = 1e9+7;
map<string,int>mp;
int qmi(int a,int b){
    int res=1;
    while(b){
        if(b&1)res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}
void init(){
    for(int i=6;i>=0;i--){
        int good=(13-i*2)*3;
        for(int j=good;j<=123;j++){
            int bad=j-good;
            int fact=qmi(j,mod-2);
            dp[i][j]=(dp[i+1][j-1]*good%mod*fact%mod+dp[i][j-1]*bad%mod*fact%mod+1)%mod;
        }
    }
}
signed main(){
    init();
    int T;
    cin>>T;
    for(int _=1;_<=T;_++){
        string s;
        cin>>s;
        mp.clear();
        int sum=0;
        string ss;
        for(int i=0;i<s.size();i+=2){
            ss="";
            ss+=s[i];
            ss+=s[i+1];
            mp[ss]++;if(mp[ss]>1)sum++;
        }
        printf("Case #%lld: %lld\n",_,dp[sum][123]);
    }
    return 0;
}