矩阵快速幂求解数列-优快云博客

本文介绍了一种使用矩阵快速幂技术高效计算特定数列的方法。通过将数列转换为矩阵运算的形式，并利用快速幂减少计算次数，该方法能够在对数时间内得到数列的第n项值。适用于解决大规模数据集上的数列计算问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Description

这里写图片描述

题解

[f n n 1] = [f n - 1 n - 1 1] \times ⎡ ⎣ ⎢ 10 k 11 011001 ⎤ ⎦ ⎥

$\begin{equation} \left[\begin{matrix} f_n&n&1 \end{matrix} \right] = \left[\begin{matrix} f_{n-1}&n-1&1 \end{matrix} \right] \times \left[\begin{matrix} 10^{k}&0&0\\ 1&1&0\\ 1&1&1 \end{matrix} \right] \end{equation}$
对于

0…9 $0\dots9$ ，

10…99 $10\dots99$ ，

100…999 $100\dots999$ ，…，

10t…n $10^{t}\dots n$ 分段计算。

#include<bits/stdc++.h>
using namespace std;

#define rep(i, l, r) for(int i = (l); i <= (r); i++)
typedef long long ll;

ll n, m;

struct Mat{
    ll dat[5][5];
    Mat(){memset(dat, 0, sizeof(dat));}
    friend Mat operator * (const Mat a, const Mat b){
        Mat mul;
        rep(i, 1, 3) rep(j, 1, 3) rep(k, 1, 3)
            mul.dat[i][j] = (mul.dat[i][j] + a.dat[i][k] * b.dat[k][j]) % m;
        return mul;
    }
    friend Mat operator ^ (const Mat p, ll n){
        Mat q = p, ret;
        rep(i, 1, 3) ret.dat[i][i] = 1;
        while(n){
            if(n & 1) ret = ret * q;
            q = q * q;
            n >>= 1;
        }
        return ret;
    }
}a;

void init(){
    scanf("%lld%lld", &n, &m);
}

void cal(ll t, ll fin){
    Mat b;
    b.dat[1][1] = t % m;
    b.dat[2][1] = b.dat[2][2] = b.dat[3][1] = b.dat[3][2] = b.dat[3][3] = 1;
    b = b ^ (fin - t / 10 + 1);
    a = a * b;
}

void work(){
    ll t = 10;
    a.dat[1][3] = 1;
    while(t <= n){
        cal(t, t-1);
        t *= 10;
    }
    cal(t, n);
    printf("%lld\n", a.dat[1][1]);
}

int main(){
    init();
    work();
    return 0;
}