I: Chiitoitsu "蔚来杯"2022牛客暑期多校训练营1
起手每种牌不超过 2 2 2 张,初始手牌里面的单牌数量可能有 1 , 3 , 5 , 7 , 9 , 11 , 13 1, 3, 5, 7, 9, 11, 13 1,3,5,7,9,11,13
定义
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j] 表示抽取
i
i
i 次牌之后剩余
j
j
j 张单牌的概率,可得状态转移方程
d
p
[
i
]
[
j
]
=
d
p
[
i
−
1
]
[
j
]
∗
123
−
(
i
−
1
)
−
3
j
123
−
(
i
−
1
)
+
d
p
[
i
−
1
]
[
j
+
2
]
∗
3
∗
(
j
+
2
)
123
−
(
i
−
1
)
dp[i][j]=dp[i-1][j]*\frac{123-(i - 1) - 3j}{123-(i-1)}+dp[i-1][j+2]*\frac{3*(j+2)}{123-(i-1)}
dp[i][j]=dp[i−1][j]∗123−(i−1)123−(i−1)−3j+dp[i−1][j+2]∗123−(i−1)3∗(j+2)
其中 124 − i − 3 j 124 − i \frac{124-i-3j}{124-i} 124−i124−i−3j 表示第 i i i 次抽牌没有抽到和手中一样类型的牌的概率, 3 ∗ ( j + 2 ) 124 − i \frac{3*(j+2)}{124-i} 124−i3∗(j+2) 表示第 i i i 次抽牌抽到了和手中单牌一样类型的牌。
总计有
123
123
123 张牌,最多抽取
121
121
121 张才能胡牌(剩下一个对子)
初始化
d
p
[
0
]
[
c
n
t
]
=
1
dp[0][cnt]=1
dp[0][cnt]=1,
c
n
t
cnt
cnt 为起手单牌数量
for(int i = 1; i<= 121; i++)
for(int j = 1;j <= 13;j += 2)
dp[i][j] = (dp[i - 1][j] * (124 - i - 3 * j) % mod + dp[i - 1][j + 2] * (3 * j + 6) % mod) % mod * qpow(124 - i, mod - 2) % mod;
最终结果
a
n
s
=
∑
i
=
1
121
i
∗
d
p
[
i
−
1
]
[
1
]
∗
3
124
−
i
ans=\sum_{i=1}^{121} i*dp[i-1][1]*\frac{3}{124-i}
ans=i=1∑121i∗dp[i−1][1]∗124−i3
所以提前把所有
c
n
t
cnt
cnt 对应的答案预处理,之后直接计算
c
n
t
cnt
cnt 输出即可。
附上完整代码
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 2e5 + 10;
const int mod = 1e9 + 7;
LL dp[130][15], res[15];
map<string, bool> hs;
inline int read() {
int x = 0, f = 1;
char ch = getchar();
for(; ch < '0' || ch > '9'; ch = getchar()) if(ch == '-') f = -1;
for(; ch >= '0' && ch <= '9'; ch = getchar()) x = (x << 1) + (x << 3) + (ch ^ 48);
return f * x;
}
LL qpow(int b, int p, int k) {
LL ans = 1, base = b;
while(p) {
if(p & 1) ans = ans * base % k;
p >>= 1;
base = base * base % k;
}
return ans % k;
}
int main() {
for(int k = 1; k <= 13; k += 2) {
memset(dp, 0, sizeof(dp));
dp[0][k] = 1;
for(int i = 1; i <= 121; i++)
for(int j = 1; j <= 13; j += 2)
dp[i][j] = (dp[i - 1][j] * (124 - i - 3 * j) % mod + dp[i - 1][j + 2] * (3 * j + 6) % mod) % mod * qpow(124 - i, mod - 2, mod) % mod;
LL ans = 0;
for(int i = 1; i <= 121; i++)
ans = (ans + i * dp[i - 1][1] * 3 % mod * qpow(124 - i, mod - 2, mod) % mod) % mod;
res[k] = ans;
}
int t = read();
for(int i = 1; i <= t; i++) {
hs.clear();
string s;
cin >> s;
int len = s.length(), cnt = 0;
for(int i = 0; i < len; i += 2) {
if(hs.find(s.substr(i, 2)) == hs.end()) {
hs[s.substr(i, 2)] = 1;
cnt++;
}
else cnt--;
}
// printf("%d\n", cnt);
printf("Case #%d: %lld\n", i, res[cnt]);
}
return 0;
}