2019牛客多校B generator 1——十进制快速幂

题目

已知 $x_i = ax_i + bx_{i-1}$，求 $x_n \% MOD$.（$1\leq n\leq 10^{(10^6)}$）

分析

写成矩阵快速幂的形式，相当于求转移矩阵的 $n$ 次幂。

由于 $n$ 过大，只能用字符串形式保存，如果转成二进制复杂度过高，就直接用十进制好了。

其实十进制快速幂和二进制几乎一样，都是倍增的思想。

ll qpow(ll a, ll b, ll p)
{
    ll ret = ;
    while(b)
    {
        if(b&) ret = ret*a%p;
        a = a*a%p;
        b >>= ;
    }
    return ret;
}

inline ll shi_pow(ll a, ll b, ll p)
{
    ll ret = ;
    while(b)
    {
        ll yu = b%;
        if(yu) ret = ret*qpow(a,yu,p)%p;
        a = qpow(a, , p);
        b /= ;
    }
    return ret;
}

二进制更快，里面能用二进制的换成了二进制。

回到题目，将字符串 $n$ 从高到低就是十进制，与上面类似

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int N=+;
ll x0,x1,a,b,mod;
char s[N];

struct Mat{
    int r,c;
    ll m[][];
    Mat(){
        //memset(m,0,sizeof(m));
        for(int i = ;i < ;i++)
            for(int j = ;j < ;j++)
                m[i][j]=;
    }
};

inline Mat mmul(Mat x,Mat y,ll p){
    Mat ans;
    ans.r=x.r;
    ans.c=y.c;
    for(int i=;i<x.r;i++)
        for(int k=;k<x.c;k++)
            for(int j=;j<y.c;j++){
                ans.m[i][j] = (ans.m[i][j] + x.m[i][k]*y.m[k][j])%p;
            }
    return ans;
}

inline Mat mpow(Mat x,ll y,ll p){
    Mat ans;
    ans.r=x.r;
    ans.c=x.c;
    for(int i=;i<ans.c;i++) ans.m[i][i]=;
    while(y){
        if(y&) ans=mmul(ans,x,p);
        x=mmul(x,x,p);
        y>>=;
    }
    return ans;
}

inline Mat m_shi_pow(Mat x,char* s,ll p){
    Mat ans;
    ans.r=x.r;
    ans.c=x.c;
    int len = strlen(s);
    for(int i=;i<ans.c;i++) ans.m[i][i]=;
    while(len--){
        int yu = (s[len]-'');  //printf("yu:%d\n", yu);
        if(yu) ans=mmul(ans,mpow(x, yu, p),p);
        x=mpow(x,,p);
    }
    return ans;
}

int main(){
    scanf("%lld%lld%lld%lld", &x0, &x1, &a, &b);
    scanf("%s%lld", s, &mod);
    Mat A,T;
    A.r=; A.c=;
    A.m[][]=x1; A.m[][]=x0;
    T.r=; T.c=;
    T.m[][]=a; T.m[][]=b; T.m[][]=; T.m[][]=;
    T = m_shi_pow(T, s, mod);
    A = mmul(T, A, mod);
    printf("%lld\n", A.m[][]);

    return ;
}

这题有点卡常，加些常数优化才抖过去。