题意:给出一个字符串a和一个整数k,表示这个字符串s是由k个 a连接得到的,要求任意删除字符串s中的任意位上的字符,使得删除后的字符串表示的数能被5整除。题目说明所求结果可能包含前导0的情况,即5和05算两种情况。
分析:要使这个数能被5整除,则删除后的字符末尾数字一定是5或0,这时只用枚举a的各位字符,如果遇到字符0或5就计算当前情况的总个数(2^i),而后面的k个串的相同位置的结果总个数分别(2^(k+len)),(2^(k+len*2))...,(2^(k+len*(k-1)),利用等比数列求和公式可以求得此时总个数为(2^n(2^(k*len))/(2^len-1)),因为分子分母都可能会爆炸,又存在对除数取模的运算,这时就需要用到快速幂和逆元的知识来解决。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<string>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<vector>
using namespace std;
typedef long long ll;
const int mod=1000000007;
const int maxn=100005;
char s[maxn];
int k;
ll pow_mod(ll a,ll b,ll m)//快速幂模版
{
ll d,t;
d=1;
t=a;
while (b>0)
{
if (b%2==1)
d=(d*t)%m;
b/=2;
t=(t*t)%m;
}
return d;
}
int main()
{
while(~scanf("%s",s))
{
if(s[0]=='-'||s[0]=='+')
{
for(int i=0;s[i];i++)
s[i]=s[i+1];
}
scanf("%d",&k);
ll sum=0;
ll len=strlen(s);
ll a=pow_mod(2,len,mod)%mod;
a=(a-1+mod)%mod;
a=pow_mod(a,mod-2,mod)%mod; //利用逆元来求a=(1/(2^len-1))%mod
ll b=pow_mod(2,len*k,mod)%mod;
b=((b%mod)-(1%mod)+mod)%mod;//b=(2^(k*len)-1)%mod
for(int i=0;i<len;i++)
{
if(s[i]=='5'||s[i]=='0')
{
ll num=1;
num=pow_mod(2,i,mod)%mod;
num=(num*b)%mod*a%mod; //num=2^i%mod
sum=(sum+num)%mod; //注意sum相加之后也可能爆炸,也要对mod取模
}
}
printf("%I64d\n",sum);
}
return 0;
}