【数论】同余（七）：快速幂、矩阵快速幂

default111

已于 2022-03-05 22:14:22 修改

阅读量175

点赞数

分类专栏：算法竞赛文章标签：算法 c++ 数学

于 2022-03-04 23:48:36 首次发布

本文链接：https://blog.csdn.net/default012/article/details/123287600

版权

算法竞赛专栏收录该内容

31 篇文章 3 订阅

订阅专栏

同余问题共7part，我的博客链接：

快速幂

普通快速幂

形式： $x^n \mod m$

原理：将 $n$ 拆成二进制，原式变为 $\begin{aligned} &x^{n_0\times 2^{0}+n_1\times 2^{1}+\cdots+n_k\times 2^{k}}\\=&x^{n_0\times 2^{0}}\cdot x^{n_1\times 2^{1}}\cdots x^{n_k\times 2^{k}}\\=&(x^{2^{0}})^{n_0}\cdot(x^{2^{1}})^{n_1}\cdots(x^{2^{k}})^{n_k} \end{aligned}$

// 快速幂：计算x^n mod m
ll qpow(ll x, ll n, ll m)
{
    ll ans = 1ll;
    while (n)
    {
        if (n & 1)
            ans = (ans * x) % m;
        x = x * x % m;
        n >>= 1ll;
    }
    return ans;
}

矩阵快速幂

形式： $n\times n$ 的矩阵 $A$ 的 $k$ 次方： $A^k$

与普通快速幂原理一样，只不过是矩阵乘法

struct matrix
{
    ll a[N][N];
    ll m;
    int n;
    matrix(int x = 0, ll mod) : n(x), m(mod) {}
    void build() // 读入矩阵
    {
        scanf("%d", &n);
        scanf("%lld", &m);
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= n; j++)
                scanf("%lld", &a[i][j]);
    }
    void init() // 矩阵初始化为单位矩阵
    {
        for (int i = 1; i <= n; i++)
            a[i][i] = 1ll;
    }
    matrix operator*(const matrix &b)
    {
        matrix nxt;
        for (int k = 1; k <= n; k++)
            for (int i = 1; i <= n; i++)
                for (int j = 1; j <= n; j++)
                    nxt.a[i][j] = (nxt.a[i][j] % m + a[i][k] % m * b.a[k][j] % m) % m;
        return nxt;
    }
};
matrix matrix_qpow(matrix A, ll k)
{
    matrix ans(A.n, A.m);
    ans.init();
    while (k)
    {
        if (k & 1)
            ans = ans * A;
        A = A * A;
        k >>= 1ll;
    }
}