1303. 斐波那契前 n 项和 矩阵乘积 矩阵快速幂

题目

题解思路

On肯定解决不了的 。

这里用到了矩阵快速幂 ， 属实是复习了一波线性代数。
有关矩阵的乘法

定义F为 1阶矩阵。其中f是feib数 S为前几项的和

构造出负责矩阵来运算
利用辅助矩阵进行矩阵快速幂使得运算复杂度变成 log级别。
参考文章

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#include <map>
#include <string>
using namespace std;

const  int  INF =  0x3f3f3f3f ;
const  int  N = 3 ;
int  n , m  ;
int A[N][N] = {
    {0,1,0},{1,1,1},{0,0,1}
};
int S[N] = {1,1,1};

void mutl(int a[N] , int b[][N] )  //一阶和三阶相乘
{
    int t[N] = {0} ;
    for (int i = 0 ; i < N ; i++ )
        for (int j = 0 ; j < N ; j++ )
            t[i] = ( t[i] + (long long )a[j]*b[j][i] ) % m ;
    for (int i = 0 ; i < N ; i++ )
        S[i] = t[i] % m ;
}


void mutl( int a[][N] , int b[][N] )   //两个三阶矩阵相乘
{
    int t[N][N] = {0} ;
    for (int i = 0 ; i < N ; i++ )
        for (int j = 0 ; j < N ; j++ )
            for (int k = 0 ; k < N ; k++ )
                t[i][j] =  ( t[i][j] + (long long )a[i][k]*b[k][j] ) % m ;
    for (int i = 0 ; i < N ; i++ )
        for (int j = 0 ; j < N ; j++ )
            A[i][j] = t[i][j] % m;
}

int main ()
{
    ios::sync_with_stdio(false);
    cin >> n >> m ;
    n-- ;
    while( n )
    {
        if ( n & 1 )
            mutl(S , A );
        mutl( A , A ) ;
        n >>= 1 ;
    }
    cout << S[2] % m <<"\n";
    return 0 ;
}