Wannafly挑战赛18 B 随机数 (十进制矩阵快速幂）

大致题意

思路

设dp[i][0] 表示前 i 次产生 偶数个1 的概率；
设dp[i][1] 表示前 i 次产生 奇数个1 的概率；
设产生1的概率为p，那么容易得到转移方程
dp[i][0] = dp[i-1][0]x(1-p) + dp[i-1][1]xp
dp[i][1] = dp[i-1][0]xp + dp[i-1][1]x(1-p)
可以矩阵加速运算，一开始我感觉这个N的范围太大，估计要T，实际上跑的还是很快的。由于N很大所以直接模拟十进制的矩阵快速幂即可，中间的每一位小数字也可以用普通快速幂算，复杂度基本上是 O(C*N)

代码

#include<bits/stdc++.h>
using namespace std;
#define maxn 1000005
#define maxm 5000006
#define ll long long int
#define INF 0x3f3f3f3f
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
#define mem(a) memset(a,0,sizeof(a))
#define sqr(x) (x*x)
#define inf (ll)2e18+1
#define mod 1000000007
int read(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
     while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
     return f*x;
}
int m,n;
char s[maxn];
struct node{int a[2][2];}e,o;
ll fast(ll x,ll y){
    ll res=1ll;
    while(y){
        if(y&1)res=res*x%mod;
        x=x*x%mod;
        y>>=1;
    }
    return res;
}
void init(){
    inc(i,0,1)inc(j,0,1){
        if(i==j)e.a[i][j]=1;
        o.a[i][j]=0;
    }
}
node operator * (node p,node q){
    node res=o;
    inc(i,0,1)inc(j,0,1)inc(k,0,1)res.a[i][j]=(1ll*res.a[i][j]+1ll*p.a[i][k]*q.a[k][j])%mod;
    return res;
}
node ksm(node p,int y){
    node res=e;
    while(y){
        if(y&1)res=res*p;
        p=p*p;
        y>>=1;
    }
    return res;
}
int main()
{
    init();
    ll inv=fast(10000,mod-2);
    scanf("%d",&m);
    n=10000-m;
    m=1ll*m*inv%mod;
    n=1ll*n*inv%mod;
    scanf("%s",s);
    int len=strlen(s);
    node pre=o,ans=e;
    pre.a[0][0]=n;pre.a[0][1]=m;pre.a[1][0]=m;pre.a[1][1]=n;
    dec(i,len-1,0){
        ans=ans*ksm(pre,s[i]-'0');
        pre=ksm(pre,10);
    }
    printf("%d\n",ans.a[1][0]);
    return 0;
}