斐波那契前 n 项和

斐波那契前 n 项和

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

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

输入格式

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

输出格式

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

数据范围

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

输入样例：

5 1000

输出样例：

12

 

解题思路

　　遇到递推的题目想想能不能构造出矩阵，然后通过矩阵乘法和快速幂来快速求出第$n$项是什么。

　　设斐波那契的第$n$项为$f_n$，定义向量$F_n = \begin{bmatrix} f_n & f_{n+1} \end{bmatrix}$，那么$F_{n+1} = \begin{bmatrix} f_{n+1} & f_{n+2} \end{bmatrix}$。尝试构造一个$2 \times 2$的矩阵$A$使得$F_n \times A = F_{n+1}$。容易发现 $$\begin{bmatrix} f_n & f_{n+1} \end{bmatrix} \times \begin{bmatrix} 0 & 1 \\ 1 & 1 \end{bmatrix} = \begin{bmatrix} f_{n+1} & f_{n+2} \end{bmatrix}$$

　　因此$A = \begin{bmatrix} 0 & 1 \\ 1 & 1 \end{bmatrix}$。同时根据矩阵乘法的结合律，有$F_n = F_1 \times A^{n-1}$，$A^{n-1}$可以用快速幂来算。

　　现在的问题是求斐波那契的前$n$项和$S_n = \sum\limits_{i=1}^{n}{f_n}$。那么我们构造向量$F_n = \begin{bmatrix} f_n & f_{n+1} & S_n \end{bmatrix}$，继续找到一个$3 \times 3$的矩阵$A$使得$F_n \times A = F_{n+1}$。同样有 $$\begin{bmatrix} f_n & f_{n+1} & S_n \end{bmatrix} \times \begin{bmatrix} 0 & 1 & 0 \\ 1 & 1 & 1 \\ 0 & 0 & 1 \end{bmatrix} = \begin{bmatrix} f_{n+1} & f_{n+2} & S_{n+1} \end{bmatrix}$$

　　那么$A = \begin{bmatrix} 0 & 1 & 0 \\ 1 & 1 & 1 \\ 0 & 0 & 1 \end{bmatrix}$。$F_n = F_1 \times A^{n-1}$，其中$F_1 = \begin{bmatrix} 1 & 1 & 1 \end{bmatrix}$。

　　AC代码如下，时间复杂度为$O\left( 3^3 \times \log{n} \right)$：

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

int n, m;
int a[3][3] = {
    {0, 1, 0},
    {1, 1, 1},
    {0, 0, 1}
};

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

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

int main() {
    cin >> n >> m;
    int f1[3] = {1, 1, 1};
    n--;
    while (n) {
        if (n & 1) mul(f1, f1, a);  // f1 = f1 * a
        mul(a, a, a);   // a = a * a
        n >>= 1;
    }
    cout << f1[2];
    
    return 0;
}

　　为了方便统一成两个矩阵的乘法，可以把$F_1$扩展为$3 \times 3$的矩阵$F1 = \begin{bmatrix} 1 & 1 & 1 \\ 0 & 0 & 0 \\ 0 & 0 & 0 \end{bmatrix}$。

　　AC代码如下：

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

int n, m;
int a[3][3] = {
    {0, 1, 0},
    {1, 1, 1},
    {0, 0, 1}
};

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

int main() {
    cin >> n >> m;
    int f1[3][3] = {1, 1, 1};
    n--;
    while (n) {
        if (n & 1) mul(f1, f1, a);
        mul(a, a, a);
        n >>= 1;
    }
    cout << f1[0][2];
    
    return 0;
}

 

参考资料

　　AcWing 1303. 斐波那契前 n 项和（算法提高课）：https://www.acwing.com/video/704/