2020icpc小米决赛K

__hzy

于 2020-12-22 22:37:30 发布

阅读量91

点赞数

分类专栏：数据结构文章标签： dfs

本文链接：https://blog.csdn.net/qq_33106167/article/details/111568798

版权

数据结构专栏收录该内容

2 篇文章 0 订阅

订阅专栏

暴力

#include<bits/stdc++.h>
using namespace std;typedef long long ll;const int mod=998244353;ll ans,g,i;string s;
ll pow_mod(ll x,ll n){
 ll res=1;
 while(n){
  if(n&1)res=res*x%mod;
  x=x*x%mod;
  n>>=1; 
 }return res;
}
bool ok(ll x){for(int i=2;i*i<=x;++i)if(x%i==0)return true;return false;}
void dfs(ll x,int len,int prob){
 if(ok(x))return ans=(ans+1LL*len*prob%mod)%mod,void();
 for(int i=0;i<s.size();++i)if(s[i]=='1')dfs(x*10+i+1,len+1,1LL*prob*g%mod);
}
int main(){
 for(cin>>s,i=0;i<s.size();++i)if(s[i]=='1')g++;g=pow_mod(g,mod-2);dfs(0,0,1);printf("%lld\n",ans);
}