【ACWing】1304. 佳佳的斐波那契

题目地址：

https://www.acwing.com/problem/content/1306/

佳佳对数学，尤其对数列十分感兴趣。在研究完Fibonacci数列后，他创造出许多稀奇古怪的数列。例如用$S(n)$表示Fibonacci前$n$项和$\mathrm{mod}m$的值，即$S_n=(F_1+F_2+\dots +F_n)\mathrm{mod}m$，其中$F_1=F_2=1,F_i=F_{i-1}+F_{i-2}$。可这对佳佳来说还是小菜一碟。终于，她找到了一个自己解决不了的问题。用$T_n=(F_1+2F_2+3F_3+\dots +n{F}_n)\mathrm{mod}m$表示Fibonacci数列前$n$项变形后的和$\mathrm{mod}m$的值。现在佳佳告诉你了一个$n$和$m$，请求出$T_n$的值。

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

输出格式：
共一行，输出$T(n)$的值。

数据范围：
$1\le n,m\le 2^{31}-1$

暂时不考虑模$m$。考虑$$\begin{gathered} P_n=n{S}_n-T_n=(n-1)F_1+(n-2)F_2+...+F_{n-1} \\ {P}_{n-1}=(n-1)S_{n-1}-T_{n-1}=(n-2)F_1+(n-3)F_2+...+F_{n-2} \end{gathered}$$所以$P_n-P_{n-1}=F_1+F_2+...+F_{n-1}=S_{n-1}$。从而$$\left[\begin{array}{c}{P}_n\\ S_n\\ F_{n+1}\\ F_n\end{array}\right]\left[\begin{array}{cccc}1& 1& 0& 0\\ 0& 1& 1& 0\\ 0& 0& 1& 1\\ 0& 0& 1& 0\end{array}\right]=\left[\begin{array}{c}{P}_{n-1}\\ S_{n-1}\\ F_n\\ F_{n-1}\end{array}\right]$$令$F_0=0$，所以$$\left[\begin{array}{c}{P}_n\\ S_n\\ F_{n+1}\\ F_n\end{array}\right]={\left[\begin{array}{cccc}1& 1& 0& 0\\ 0& 1& 1& 0\\ 0& 0& 1& 1\\ 0& 0& 1& 0\end{array}\right]}^n\left[\begin{array}{c}0\\ 0\\ 1\\ 0\end{array}\right]$$问题归结为求$P_n$和$S_n$，从而可以用快速幂来做。代码如下：

#include <iostream>
using namespace std;

using ll = long long;

int n, m;
ll a[4][4], res[4][4];
void mult(ll a[][4], ll b[][4]) {
  ll c[4][4] = {0};
  for (int i = 0; i < 4; i++)
    for (int j = 0; j < 4; j++)
      for (int k = 0; k < 4; k++) c[i][j] = (c[i][j] + a[i][k] * b[k][j]) % m;
  for (int i = 0; i < 4; i++)
    for (int j = 0; j < 4; j++) a[i][j] = c[i][j];
}

void fast_pow(int n) {
  while (n) {
    if (n & 1) mult(res, a);
    n >>= 1;
    mult(a, a);
  }
}

int main() {
  scanf("%d%d", &n, &m);
  for (int i = 0; i < 4; i++) res[i][i] = 1;
  a[0][0] = a[0][1] = a[1][1] = a[1][2] = a[2][2] = a[2][3] = a[3][2] = 1;
  fast_pow(n);
  ll pn = res[0][2], sn = res[1][2];
  printf("%lld\n", ((n * sn - pn) % m + m) % m);
}

时间复杂度$O(\log n)$，空间$O(1)$。