[bzoj 2326--HNOI2011]数学作业

小C数学成绩优异，于是老师给小C留了一道非常难的数学作业题：给定正整数 N 和 M 要求计算 Concatenate (1 ..N) Mod M 的值，其中 Concatenate (1 ..N)是将所有正整数 1, 2, …, N 顺序连接起来得到的数。例如，N> = 13, Concatenate (1 .. N)=12345678910111213.小C 想了大半天终于意识到这是一道不可能手算出来的题目， 于是他只好向你求助，希望你能编写一个程序帮他解决这个问题。

蒟蒻百篇博客纪念！
这道题看到这么大，便不难想到要快速幂，之后再看到它的相乘时这么有规律，便是矩乘。为什么说有规律，因为f[i]=f[i-1]*10^k+i(f[i]表示Concatenate (1 .. i)，k表示i的位数)，这个式子如此单一，便可以加速。但是10^k这个在各个位数上不一样啊，那就分各个位数来做啦，之后就非常容易了。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
using namespace std;
long long mod;
struct node
{
    long long a[5][5];
    node(){memset(a,0,sizeof(a));}
};
node chengfa1(node a,node b)
{
    node c;
    for(int i=1;i<=3;i++)
    {
        for(int j=1;j<=3;j++)
        {
            for(int k=1;k<=3;k++)
            {
                c.a[i][j]=(c.a[i][j]+a.a[i][k]*b.a[k][j])%mod;
            }
        }
    }
    return c;
}
node chengfa2(node a,node b)
{
    node c;
    for(int i=1;i<=3;i++)
    {
        for(int k=1;k<=3;k++)
        {
            c.a[i][1]=(c.a[i][1]+a.a[i][k]*b.a[k][1])%mod;
        }
    }
    return c;
}
long long power1(long long a,long long b)
{
    long long ans=1;
    while(b)
    {
        if(b%2==1)ans=ans*a;
        a=a*a;
        b/=2;
    }
    return ans;
}
node power2(node a,long long b)
{
    node ans;
    ans.a[1][1]=ans.a[2][2]=ans.a[3][3]=1;
    while(b)
    {
        if(b%2==1)ans=chengfa1(a,ans);
        a=chengfa1(a,a);
        b/=2;
    }
    return ans;
}
int main()
{
    long long n,s=0,ss;
    scanf("%lld%lld",&n,&mod);ss=n;
    while(ss)ss/=10,s++;
    node pre,per;
    pre.a[1][2]=pre.a[1][3]=pre.a[2][2]=pre.a[2][3]=pre.a[3][3]=1;
    per.a[3][1]=1;
    for(int i=1;i<=s;i++)
    {
        pre.a[1][1]=power1(10,i)%mod;
        if(i!=s)per=chengfa2(power2(pre,power1(10,i-1)*9),per);
        else per=chengfa2(power2(pre,(n-power1(10,s-1)+1)),per);
    }
    printf("%lld\n",per.a[1][1]);
    return 0;
}