CodeForces 327C Magic Five 题解&代码

其实挺水的？【并不
反正代码不长

思路：
对于一个串，我们很容易知道对于某一位i是0或5，有2^i种选法【字符串从第0位开始，第i位确定】
一个串的选法也很容易求出，按顺序求出每一位的选法，加起来【设一个串有k种选法
对于多个串，第二个串的某一位i【对于该串的第i位】，有2^(strlen(s)+i)种选法，即(2^i)×(2^strlen(s))种选法。这样的话，选法之和可以提取为(2^strlen(s))k种选法
这样的话，m个串的选法之和可以整理为等比数列之和，等比数列公式为ai=k(2^(strlen(s)×(i-1)))
按照等比数列求和公式我们得到m个串的和Tm=(1-2^(strlen(s)*m))/(1-2^strlen(s))%mod

然后咧，数学公式推完该算了
除法取模= =嗯，第一反应是求逆元…关于逆元的求法我刚又忘了【查了一发】，费马小定理求a的逆元b：
ab≡1(mod p)
a^(p-1)≡1(mod p)【p是素数】
逆元b就是a^(p-2)，非常机智的转化

快速幂求Tm

没了

**少取模了一次WA了好几发，后来查数据发现自己有算出负数才反应上来

#include<iostream>
#include<stdio.h>
#include<string.h>
#define MOD 1000000007
#define LL long long
using namespace std;
int k,l,x,y;
LL cnt;
string a;
LL pow(LL a,LL b)
{
    LL r=1;
    while(b)
    {
        if(b&1)r*=a,r%=MOD;
        a*=a,a%=MOD;
        b>>=1;
    }
    return r;
}
int main(void)
{
    cin>>a>>k;
    l=a.size();
    for(int i=0;i<l;i++)
        if(a[i]=='0' || a[i]=='5')cnt+=pow(2,i),cnt%=MOD;
    x=pow(2,l);
    y=pow(x,k);
    x=((1-x)%MOD+MOD)%MOD;
    y=((1-y)%MOD+MOD)%MOD;
    cout<<(cnt*((y*pow(x,MOD-2))%MOD))%MOD<<endl;
    return 0;
}