1303. 斐波那契前 n 项和(矩阵快速幂)

大家都知道 Fibonacci 数列吧，f1=1,f2=1,f3=2,f4=3,…,fn=fn−1+fn−2

现在问题很简单，输入 n 和 m，求 fn 的前 n� 项和 Sn mod m

输入格式

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

输出格式

输出前 n 项和 Sn mod m 的值。

数据范围

1≤n≤2000000000
1≤m≤1000000010

输入样例：

5 1000

输出样例：

12

#include <iostream>
#include <cstring>
using namespace std;
typedef long long ll;
constexpr int N = 3, INF = 2e9;
int n, mod;
int f1[3] = {1, 1, 1};
int A[N][N] = {
        {0, 1, 0},
        {1, 1, 1},
        {0, 0, 1},

};
void mul(int c[], int a[], int b[][N])
{
    int temp[N] = {0};
    for (int i = 0; i < N; i++) // i代表的是坐标
    {
        for (int j = 0; j < N; j++) // j代表操作；
        {
            temp[i] = (temp[i] + (ll)a[j] * (ll)b[j][i]) % mod;
        }
    }
    memcpy(c, temp, sizeof temp);
}
void mul(int c[][N], int a[][N], int b[][N])
{
    int temp[N][N] = {0};
    for (int i = 0; i < N; i++)
    {
        for (int j = 0; j < N; j++)
        {
            for (int k = 0; k < N; k++)
            {
                temp[i][j] = (temp[i][j] + (ll)a[i][k] * b[k][j]) % mod;
            }
        }
    }
    memcpy(c, temp, sizeof temp);
}
int main()
{
    cin >> n >> mod;
    n--;
    while (n)
    {
        if (n & 1)
        {
            mul(f1, f1, A);
        }
        mul(A, A, A);
        n >>= 1;
    }
    cout << f1[2] << endl;
    return 0;
}