2020 ICPC·小米邀请赛 决赛 Rikka with Composite Number

原题链接

题意

1. 给定一个集合D{1，2，3，4，5，6，7，8，9}，一个数C=0，执行操作 C=C10+d（集合D中的一个数，选中的概率相同），直到C是合数（除1和其本身，还有别的因子）停止，求一个值res，就是所有合数的概率操作次数
2. 模拟样例1：输入100000000 ，代表集合D{1}，选1的概率是1 。第一步：c=010+1=1（不是合数）;第二步：c=110+1=11(不是合数);第三步：c=1110+1=111(合数) 所以最终结果是11*1 *3 。3是操作次数，1是概率
3. 模拟样例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是概率
4. 最后再对结果取模即可

思路

1. 这场比赛的签到题，全是模板的套用，直接模拟就好，用到快速幂，试除法判定质数，dfs
2. 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;
}