bzoj 2326: [HNOI2011]数学作业

solution

矩阵：

f：                         a：

                                   10^k    0       0

ans    i    1    *                1       1       0

                                      1       1       1

分段乘，给a赋值10^k时记得给10^k%上mod，不然炸long long了

#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long
#define chua(k) a[1][1]=p[k]%mod;a[1][2]=a[1][3]=a[2][3]=0;a[2][1]=a[2][2]=a[3][2]=a[3][3]=a[3][1]=1
using namespace std;

ll n;
ll mod,weishu;
ll f[5],a[5][5],temp[5][5];
ll p[101],ci[101];

void out11()
{
    printf("
");
    for(int i=1;i<=3;++i)
      printf("%lld ",f[i]);
    printf("
");
    for(int i=1;i<=3;++i)
    {
        for(int j=1;j<=3;++j)
          printf("%lld ",a[i][j]);
        printf("
");
    }
    printf("
");
}

void work()
{
    p[1]=10;ci[1]=9;
    for(int i=2;i<=18;++i)
    {
      p[i]=p[i-1]*10;
      ci[i]=ci[i-1]*10;
      //cout<<ci[i]<<endl;
    }
    
    for(ll i=n;i;i/=10)
      ++weishu;
    
    ci[weishu]=n-p[weishu-1]+1;
    
    //cout<<weishu<<endl;
    
    //for(int i=2;i<=18;++i)
    //  cout<<ci[i]<<endl;
    
    
    
    //cout<<weishu<<endl;
    
    f[1]=0;f[2]=0;f[3]=1;
    for(int l=1;l<=weishu;++l)
    {
        chua(l);
        
        while(ci[l])
        {
            //out11();
            
            if(ci[l]&1)
            {
                for(int i=1;i<=3;++i)
                {
                    temp[1][i]=0;
                    for(int k=1;k<=3;++k)
                        //temp[1][i]=(temp[1][i]+f[k]*a[k][i]);
                        temp[1][i]=(temp[1][i]+f[k]*a[k][i]%mod)%mod;
                }
                for(int i=1;i<=3;++i)
                  f[i]=temp[1][i];
            }
            
            for(int i=1;i<=3;++i)
              for(int j=1;j<=3;++j)
              {
                  temp[i][j]=0;
                  for(int k=1;k<=3;++k)
                      temp[i][j]=(temp[i][j]+a[i][k]*a[k][j]%mod)%mod;
                    //temp[i][j]=(temp[i][j]+a[i][k]*a[k][j]);
              }
            for(int i=1;i<=3;++i)
              for(int j=1;j<=3;++j)
                a[i][j]=temp[i][j];
            
            ci[l]>>=1;
        }
    }
}

int main(){
    
//    freopen("1.txt","r",stdin);
    //freopen("2.txt","w",stdout);
    
    scanf("%lld%lld",&n,&mod);
    work();
    printf("%lld",f[1]);
    return 0;
}