BZOJ 2326 数学作业(分段矩阵快速幂)

实际上，对于位数相同的连续段，可以用矩阵快速幂求出最后的ans，那么题目中一共只有18个连续段。

分段矩阵快速幂即可。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<cmath>
#define ll long long
using namespace std;
const int maxn=;
ll n,m;
struct Mat
{
    ll mat[maxn][maxn];
    Mat()
    {
        memset(mat,,sizeof(mat));
    }
    Mat operator *(const Mat &b)
    {
        Mat c;
        for(int k=;k<=;k++)
        {
            for(int i=;i<=;i++)
            {
                for(int j=;j<=;j++)
                {
                    c.mat[i][j]+=mat[i][k]*b.mat[k][j];
                    c.mat[i][j]%=m;
                }
            }
        }
        return c;
    }  

};
Mat ans;
Mat cal(ll k,ll t)
{
    Mat res;
    Mat b;
    ll y=t-k/+;
    b.mat[][]=k%m;
    b.mat[][]=b.mat[][]=b.mat[][]=b.mat[][]=b.mat[][]=;
    for(int i=;i<=;i++)res.mat[i][i]=;
    while(y)  {
        if(y&)res=res*b;
        y>>=;
        b=b*b;
    }
    return res;
}
int main()
{
    ans.mat[][]=;
    scanf("%lld%lld",&n,&m);
    ll t=;
    for(int i=;i<=;i++) {
        ans=cal(t,t-)*ans;
        t*=;
        if(t>n)break;
    }
    ans=cal(t,n)*ans;
    printf("%lld\n",ans.mat[][]);
    return ;
}