codeforces-327C Magic Five（等比数列求和+快速幂+逆元）

点击打开题目链接

题意：给出一个字符串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;
}