矩阵快速幂 - 斐波那契前 n 项和 - AcWing 1303

最新推荐文章于 2024-03-03 18:15:29 发布

njuptACMcxk

最新推荐文章于 2024-03-03 18:15:29 发布

阅读量606

点赞数 2

分类专栏：矩阵文章标签：算法矩阵快速幂数论 ACM 数学

本文链接：https://blog.csdn.net/njuptACMcxk/article/details/107360151

版权

矩阵专栏收录该内容

9 篇文章

订阅专栏

矩阵快速幂 - 斐波那契前 n 项和 - AcWing 1303

$大家都知道 Fibonacci 数列吧，f_1=1,f_2=1,f_3=2,f_4=3,…,f_n=f_{n−1}+f_{n−2}。$

$现在问题很简单，输入 n 和 m，求 f_n 的前 n 项和 S_n\ mod\ m。$

输入格式

$共一行，包含两个整数 n 和 m 。$

输出格式

$输出前 n 项和 S_n\ mod\ m 的值。$

数据范围

$1 \leq n \leq 2000000000, 1 \leq m \leq 1000000010$

输入样例：

5 1000

输出样例：

分析：

$设F_n=[f_n\quad f_{n+1}\quad S_n]，$

$配凑3×3的矩阵A，使得F_nA=F_{n+1}。$

$即[f_n\quad f_{n+1}\quad S_n]\begin{vmatrix}\qquad\qquad\qquad\\\qquad\\\qquad\end{vmatrix}=[f_{n+1}\quad f_{n+2}\quad S_{n+1}]$

$f_{n+1}=0×f_{n}+1×f_{n+1}+0×S_n$

$f_{n+2}=1×f_{n}+1×f_{n+1}+0×S_n$

$S_{n+1}=0×f_{n}+1×f_{n+1}+1×S_n$

$配凑出A=\begin{vmatrix}0 \quad1\quad0\\1 \quad1\quad0\\0 \quad1\quad1\end{vmatrix}。$

$则同过矩阵乘法：F_1A^{n-1}=F_n=[f_n\quad f_{n+1}\quad S_n]，计算得到F_n[2]=S_n。$

代码：

#include<iostream>
#include<cstring>

#define ll long long

using namespace std;

const int N=3;

int n,m;
ll f1[N][N]={{1,1,1}};
ll A[N][N]={
    {0,1,0},
    {1,1,1},
    {0,0,1}
};

void mul(ll a[][N],ll b[][N])
{
    ll c[N][N]={0};
    for(int i=0;i<3;i++)
        for(int j=0;j<3;j++)
            for(int k=0;k<3;k++)
                c[i][j]=(c[i][j]+a[i][k]*b[k][j]%m)%m;
    memcpy(a,c,sizeof c);
}

void quick_pow(ll a[][N],int k)
{
    ll E[N][N]={    //单位阵
        {1,0,0},
        {0,1,0},
        {0,0,1}
    };
    while(k)
    {
        if(k&1) mul(E,a);
        mul(a,a);
        k>>=1;
    }
    memcpy(a,E,sizeof E);
}

int main()
{
    cin>>n>>m;
    
    quick_pow(A,n-1);
    mul(f1,A);
    
    cout<<f1[0][2]<<endl;
    
    return 0;
}