矩阵快速幂 UVA 10870 Recurrences

最新推荐文章于 2020-04-12 23:40:54 发布

weixin_34219944

最新推荐文章于 2020-04-12 23:40:54 发布

阅读量75

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Running-Time/p/5365100.html

本文介绍了一种使用矩阵快速幂解决线性递推数列的方法，并提供了完整的C++代码实现。通过构造特定的矩阵并利用快速幂算法，可以高效地求解数列的第n项对m取模的结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目传送门

题意：f(n) = a1f(n − 1) + a2f(n − 2) + a3f(n − 3) + . . . + adf(n − d), for n > d，求f (n) % m。训练指南的题目

分析：令： $f(n-1) = \begin{bmatrix} \\ f(n-1-d) \\ f(n-2) \\ ... \\ f(n-1) \end{bmatrix}$ ， $matrix A = \begin{bmatrix} & & & \\ 0,1,0,.........0 & & & \\ 0,0,1,.........0 & & & \\ ..................... & & & \\ 0,0,0,.........1 & & & \\ ...a_{d}, a_{d-1}, a_{d-2},...a_{1} \end{bmatrix}$ .则 $F(n) = A * F(n-1) = A^{^{n-d}} * F (d)$

#include <bits/stdc++.h>

int d, n, m;
int a[16], f[16];

struct Mat {
    int m[17][17];
    int row, col;
    Mat() {
        //row = col = 16;
        memset (m, 0, sizeof (m));
    }
    void init(int sz) {
        row = col = sz;
        for (int i=1; i<row; ++i) {
            m[i][i+1] = 1;
        }
        int c = sz - 1;
        for (int i=2; i<=col; ++i) {
            m[sz][i] = a[c--];
        }
    }
    void change(int sz) {
        row = col = sz;
        for (int i=1; i<=sz; ++i) {
            m[i][i] = 1;
        }
    }
};

Mat operator * (const Mat &a, const Mat &b) {
    Mat ret;
    ret.row = a.row; ret.col = b.col;
    for (int i=1; i<=a.row; ++i) {
        for (int j=1; j<=b.col; ++j) {
            for (int k=1; k<=a.col; ++k) {
                int &r = ret.m[i][j];
                r = (r + 1ll * a.m[i][k] * b.m[k][j]) % m;
            }
        }
    }
    return ret;
}

Mat operator ^ (Mat x, int n) {
    Mat ret; ret.change (d+1);
    while (n) {
        if (n & 1) {
            ret = ret * x;
        }
        x = x * x; n >>= 1;
    }
    return ret;
}

//Running_Time
int main() {
    while (scanf ("%d%d%d", &d, &n, &m) == 3) {
        if (!d && !n && !m) {
            break;
        }
        for (int i=1; i<=d; ++i) {
            scanf ("%d", a+i);
        }
        for (int i=1; i<=d; ++i) {
            scanf ("%d", f+i);
        } 
        if (n <= d) {
            printf ("%d\n", f[n] % m);
        } else {
            Mat ans, Fd;
            ans.init (d + 1);
            ans = ans ^ (n - d);
            
            Fd.row = d + 1; Fd.col = 1;
            for (int i=2; i<=d+1; ++i) {
                Fd.m[i][1] = f[i-1];
            }

            ans = ans * Fd;
            printf ("%d\n", ans.m[d+1][1]);
        }
    }

    return 0;
}

转载于:https://www.cnblogs.com/Running-Time/p/5365100.html