hd4565 / 2013 ACM-ICPC长沙赛区全国邀请赛 A. So Easy! · 二阶齐次递推 + 矩阵快速幂

题解

大佬的完整题解 - 以及解题的关键步骤：特征法

图片来自大佬博客↓

之前一直想不通，为什么等式 $\lceil (a+\sqrt{b}{)}^n\rceil =(a+\sqrt{b}{)}^n+(a-\sqrt{b}{)}^n$ 成立，它不会超吗？现在有点想明白了，

假设$(a+\sqrt{b}{)}^n$ 的小数部分是小于0.5的，$(a-\sqrt{b}{)}^n$的小数部分是大于0.5的，小数部分相加大于1，

但这是不可能的，因为无论n是奇数还是偶数，公式 $(a+\sqrt{b}{)}^n+(a-\sqrt{b}{)}^n$ 都是整数，

而如果为了凑整，使得小数部分相加等于2，这也是不可能的，因为 $0<(a-\sqrt{b}{)}^n<1$，

故，该记一记了 $\lceil (a+\sqrt{b}{)}^n\rceil =(a+\sqrt{b}{)}^n+(a-\sqrt{b}{)}^n$

斐波那契数列通项公式推导及其类似构造共轭公式反求递推式

当有 $f_n=(a+\sqrt{b}{)}^n+(a-\sqrt{b}{)}^n$ 时

存在递推公式：$f_n=2a\ast f_{n-1}+(b-a^2)\ast f_{n-2}$

所以，如果是求某个 $f_n$ 确切的值时，要么找循环节，要么用矩阵快速幂

矩阵快速幂的转移公式：

$\left[\begin{array}{c}{f}_2\\ f_1\end{array}\right]=\left[\begin{array}{cc}2a& b-a^2\\ 1& 0\end{array}\right]\cdot \left[\begin{array}{c}{f}_1\\ f_0\end{array}\right]$

$\left[\begin{array}{c}{f}_n\\ f_{n-1}\end{array}\right]={\left[\begin{array}{cc}2a& b-a^2\\ 1& 0\end{array}\right]}^{n-1}\cdot \left[\begin{array}{c}{f}_1\\ f_0\end{array}\right]$

再手动算出$f_1$和$f_0$，就可以了

---

---

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6 + 10;
const int INF = 0x3f3f3f3f;
int n, m, k;
int a, b;
namespace MAT {//矩阵快速幂板子
    const int maxn = 2;

    struct Mat {
        ll mat[maxn][maxn];
        
        void clear() {
            memset(mat, 0, sizeof(mat));
        }

        Mat operator*(const Mat a) const {
            Mat b;
            b.clear();
            for (int i = 0; i < maxn; ++i) {
                for (int j = 0; j < maxn; ++j) {
                    for (int k = 0; k < maxn; ++k) {
                        b.mat[i][j] = (b.mat[i][j] + (mat[i][k] * a.mat[k][j]) % m + m) % m;
                    }
                }
            }
            return b;
        }
    };

    Mat pow(Mat m, ll k) {
        Mat res;
        res.clear();

        for (int i = 0; i < maxn; ++i) {//单位矩阵
            res.mat[i][i] = 1;
        }
        while (k) {
            if (k & 1)res = res * m;
            k >>= 1;
            m = m * m;
        }
        return res;
    }

    Mat E;//转移矩阵
    void init() {
        E.clear();
        /*
            fn = a*f_n-1 + b*f_n-2
            |  f  |=|a b|*|f_n-1|
            |f_n-1| |1 0| |f_n-2|

            需要一个转移矩阵 E = {a, b , 1, 0};
            以及最开始的两个f1 f0

            Mat f = {1, 0, 1, 0};// f1 f0
         */
        E = {2 * a, b - a * a, 1, 0};
    }
}
using namespace MAT;


int main() {
    ios::sync_with_stdio(0);

    while (cin >> a >> b >> n >> m) {
        init();
        Mat f = {2 * a, 0, 1, 0};// f1 f0
        f = pow(E, n - 1) * f;
        cout << f.mat[0][0] << endl;
    }
    return 0;
}