bzoj2326: [HNOI2011]数学作业（矩阵乘法）

题目传送门
。

解法：
矩阵乘法啊。
设当前数为x。要加上的为i。
若i为1为数。
那么x应该变为x*10+i。i应该变成i+1
这个过程很明显是矩阵乘法嘛。

初始矩阵，因为有个+1，所以多一个元素表示1
X i 1
转移矩阵
10 0 0
1 1 0
0 1 1
这样就变成了
X*10+i i+1 1
然后100，1000，10000同理即可。
注意指数也要用longlong。

代码实现：

#include
#include
#include
#include
#include
using namespace std;
typedef long long ll;
struct node {ll a[4][4];node() {memset(a,0,sizeof(a));}}per;
ll mod;
node jc(node a,node b,int n,int m,int p) {
    node c;
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)for(int k=1;k<=p;k++)
        c.a[i][j]=(c.a[i][j]+a.a[i][k]*b.a[k][j])%mod;
    return c;
}
node pow_mod(node a,ll b) {
    node ans=per;
    while(b!=0) {if(b%2==1)ans=jc(ans,a,3,3,3);a=jc(a,a,3,3,3);b/=2;}
    return ans;
}
int main() {
    //freopen("homework.in","r",stdin);freopen("homework.out","w",stdout);
    ll n;scanf("%lld%d",&n,&mod);
    int len=0;ll x=n;
    for(int i=1;i<=3;i++)per.a[i][i]=1ll;
    while(x!=0) {x/=10;len++;}
    ll st=1ll,ed=1ll;node ans;
    ans.a[1][1]=0ll;ans.a[1][2]=1ll;ans.a[1][3]=1ll;
    node A;
    for(int i=1;i10ll;
        A.a[1][1]=ed%mod;
        A.a[2][1]=1ll;A.a[2][2]=1ll;
                      A.a[3][2]=1ll;A.a[3][3]=1ll;
        ans=jc(ans,pow_mod(A,ed-st),1,3,3);
        st*=10ll;
    }
    ed*=10ll;
    A.a[1][1]=ed%mod;
    A.a[2][1]=1ll;A.a[2][2]=1ll;
                  A.a[3][2]=1ll;A.a[3][3]=1ll;
    ans=jc(ans,pow_mod(A,n-st+1),1,3,3);
    printf("%lld\n",ans.a[1][1]);
    return 0;
}