ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

Educational DP Contest R - Walk(倍增floyd,矩阵快速幂)

2021-08-03 11:03:05  阅读:202  来源: 互联网

标签:Educational mat Contest int ll Walk ++ sum Matrix


题目来源:AtCoder EDU DP题集

题目链接:Here

单独拎出来是因为这道题是一个很好的板子,值得记录

题意

给定一个 n 个节点的有向图的邻接矩阵,求该有向图中长度为 k 的路径长。

解法

算法涉及:倍增 Floyd

答案为该邻接矩阵的 \(k\) 次幂的行列式。

学过离散数学的后面图论的话大概都知道求有向图中长度为 \(k\) 的路径长的路径与原始图的 \(k\) 次方相关,所以只需要求原矩阵的 \(k\) 次幂即可

使用矩阵快速幂即可,时间复杂度 \(\mathcal{O}(n^2log k)\)

const int mod = 1e9 + 7;
ll n, k;
struct Matrix {
    ll mat[50][50];
    void clear() {memset(mat, 0, sizeof(mat));}
    void reset(int n) {
        clear();
        for (int i = 0; i < n; ++i) mat[i][i] = 1;
    }
} a;
Matrix MatrixMul(Matrix a, Matrix b) { // 矩阵快速乘
    Matrix t; t.clear();
    for (int i = 0; i < n; ++i)
        for (int k = 0; k < n; ++k)
            for (int j = 0; j < n; ++j)
                t.mat[i][j] = (t.mat[i][j] + a.mat[i][k] * b.mat[k][j]) % mod;
    return t;
}
ll MatrixQpow(Matrix a, ll p) { // 矩阵快速幂
    Matrix s; s.reset(n);
    for (; p; p >>= 1, a = MatrixMul(a, a))
        if (p & 1) s = MatrixMul(s, a);
    ll sum = 0;
    for (int i = 0; i < n; ++i) for (int j = 0; j < n; ++j)
            sum = (sum + s.mat[i][j]) % mod;
    return sum;
}
int main() {
    cin.tie(nullptr)->sync_with_stdio(false);
    cin >> n >> k;
    for (int i = 0; i < n; ++i) for (int j = 0; j < n; ++j) cin >> a.mat[i][j];
    cout << MatrixQpow(a, k);
}

标签:Educational,mat,Contest,int,ll,Walk,++,sum,Matrix
来源: https://www.cnblogs.com/RioTian/p/15093314.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有