斐波那契前 n 项和acwing

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

现在问题很简单，输入 n 和 m，求 fn 的前 n 项和 Snmodm。

输入格式
共一行，包含两个整数 n 和 m。

输出格式
输出前 n 项和 Snmodm 的值。

数据范围
1≤n≤2000000000,
1≤m≤1000000010
输入样例：
5 1000
输出样例：
12

解析:
Fn = [ f n f n + 1 S n ]
[fnfn+1Sn]
[fn​fn+1​Sn​]
Fn+1 = [ f n + 1 f n + 2 S n + 1 ]
[fn+1fn+2Sn+1]

[fn+1​fn+2​Sn+1​]

[ f n f n + 1 S n ]
[fnfn+1Sn]
[fn​fn+1​Sn​]* [ 0 1 0 1 1 1 0 0 1 ]
⎡⎣⎢010110011⎤⎦⎥
⎣⎡​010​110​011​⎦⎤​ = [ f n + 1 f n + 2 S n + 1 ]
[fn+1fn+2Sn+1]
[fn+1​fn+2​Sn+1​]
这要推出系数矩阵即可。

#include<bits/stdc++.h>
using namespace std;
int n, m;
void mul(int c[], int a[], int b[][3])
{
	int temp[3] = { 0 };
	for (int i = 0; i < 3; i++)
	{
		for (int j = 0; j < 3; j++)
		{
			temp[i] = (temp[i] + (long long)a[j] * b[j][i]) % m;
		}
	}
	memcpy(c, temp, sizeof temp);
}
void mul(int c[][3], int a[][3], int b[][3])
{
	int temp[3][3] = { 0 };
	for (int i = 0; i < 3; i++)
		for (int j = 0; j < 3; j++)
		{
			for (int k = 0; k < 3; k++)
			{
				temp[i][j] = (temp[i][j] + (long long)a[i][k] * b[k][j] % m) % m;
			}
		}
	memcpy(c, temp, sizeof temp);
}
int main()
{
	int f[3] = { 1,1,1 };
	int a[3][3] = {
	  {0,1,0},{1,1,1},{0,0,1}
	};
	cin >> n >> m;
	while (n)
	{
		if (n & 1)mul(f, f, a);
		mul(a, a, a);
		n /= 2;
	}
	cout << f[2] << endl;
}