BZOJ2326 数学作业

题目传送门

Solution

设N的答案为f(n)。那么很容易得到一个递推式：$F(n)=F(n-1)*10^k+n$，其中k是n的位数。
所以当k固定时，这个式子的转移方式也是固定的。所以可以枚举每个k。而对于每一个k，我们很容易得到一个矩阵：
$[10^k,1,1]$
$[0, \phantom,\phantom,\phantom,1,1]$
$[0,\phantom,\phantom,\phantom,1,1]$
对每一个k,就可以快速计算答案。

code

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
ll mo;
ll f[10][10],c[10][10],ans[10][10];
void calc1()
{
    for (int i=1;i<=3;i++)
        for (int j=1;j<=3;j++)
            c[i][j]=0;
    for (int i=1;i<=3;i++)
        for (int j=1;j<=3;j++)
            for (int k=1;k<=3;k++)
                (c[i][j]+=f[i][k]*ans[k][j]%mo)%=mo;
    for (int i=1;i<=3;i++)
        for (int j=1;j<=3;j++)
            ans[i][j]=c[i][j];
}
void calc2()
{
    for (int i=1;i<=3;i++)
        for (int j=1;j<=3;j++)
            c[i][j]=0;
    for (int i=1;i<=3;i++)
        for (int j=1;j<=3;j++)
            for (int k=1;k<=3;k++)
                (c[i][j]+=f[i][k]*f[k][j]%mo)%=mo;
    for (int i=1;i<=3;i++)
        for (int j=1;j<=3;j++)
            f[i][j]=c[i][j];
}
void ksm(ll t,ll last)
{
    memset(f,0,sizeof(f));
    f[1][1]=t%mo;f[1][2]=1;f[1][3]=1;
    f[2][2]=1;f[2][3]=1;
    f[3][3]=1;
    ll b=last-t/10+1;
    while (b)
    {
        if (b&1) calc1();
        calc2();
        b>>=1;
    }
}
int main()
{
    ll n;
    scanf("%lld%lld",&n,&mo);
    ll t=10;
    for (int i=1;i<=3;i++)
        ans[i][i]=1;
    while (t<=n)
    {
        ksm(t,t-1);
        t*=10;
    }
    ksm(t,n);
    printf("%lld\n",ans[1][3]);
    return 0;
}