AcWing 1303. 斐波那契前 n 项和
原题链接
中等
作者:
黄亦玫
,
2020-09-15 09:20:29
,
所有人可见
,
阅读 517
记录一下踩的坑
#include <iostream>
#include <cstring>
using namespace std;
typedef long long LL;
const int N = 3;
int n, m;
void mul(int f[], int a[][N])
{
int temp[N] = {0};
for(int i = 0; i < 3; i ++)
for(int j = 0; j < 3; j ++)
temp[i] = (temp[i] + (LL)f[j] * a[j][i]) % m;
memcpy(f, temp, sizeof temp); // 一定要是sizeof(temp)不能是f因为f传进来的是个指针大小就是指针的大小。
}
void mul(int a[][N], int b[][N])
{
int temp[N][N] = {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] + (LL)a[i][k] * b[k][j]) % m;
memcpy(a, temp, sizeof temp);//一定要是sizeof(temp),不能上f因为f传进来的是指针大小就是指针的大小。
}
int main()
{
cin >> n >> m;
int f[N] = {0, 1, 0};
int a[N][N] = {
{0, 1, 0},
{1, 1, 1},
{0, 0, 1},
};
while(n)
{
if(n & 1)mul(f, a);
mul(a, a);
n >>= 1;
}
cout << f[2] << endl;
return 0;
}