原题链接
题意
- 给定一个集合D{1,2,3,4,5,6,7,8,9},一个数C=0,执行操作 C=C10+d(集合D中的一个数,选中的概率相同),直到C是合数(除1和其本身,还有别的因子)停止,求一个值res,就是所有合数的概率操作次数
- 模拟样例1:输入100000000 ,代表集合D{1},选1的概率是1 。第一步:c=010+1=1(不是合数);第二步:c=110+1=11(不是合数);第三步:c=1110+1=111(合数) 所以最终结果是11*1 *3 。3是操作次数,1是概率
- 模拟样例2:输入001100000,代表集合D{3,4},选3,4的概率都是1/2 ,第一步可以选4,res=1/21
也可以选3,第二步选3,4的都可以变成33,34,都是合数,所以res+=1/22=3/2 。2是操作次数,1/2是概率 - 最后再对结果取模即可
思路
- 这场比赛的签到题,全是模板的套用,直接模拟就好,用到快速幂,试除法判定质数,dfs
- dfs直接搜索找到合数,每次累加结果即可
AC代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 998244353;
const int mi = 998244351;
char s[15];
ll res,sum;
bool is_prime(ll x) {
if (x == 1) return true;
for (int i = 2; i <= x / i; i++)
if (x % i == 0)
return false;
return true;
}
ll qmi(ll m, ll k, ll p) {
ll res = 1;
while (k) {
if (k & 1) res = res * m % p;
m = m * m % p;
k >>= 1;
}
return res;
}
void dfs(ll x,ll d){
for(int i=1;i<=9;i++){
if(s[i]=='1'){
if(is_prime(x*10+i)){
dfs(x*10+i,d+1);
}else{
res=(res+d*qmi(qmi(sum,d,mod),mi,mod))%mod;
}
}
}
}
int main() {
scanf("%s",s+1);
sum=0;
for(int i=1;i<=9;i++){
if(s[i]=='1'){
sum++;
}
}
dfs(0,1);
cout<<res<<endl;
return 0;
}