POJ 3613 Floyd + 矩阵快速幂

最新推荐文章于 2021-03-04 14:24:44 发布

SHOHOKUKU

最新推荐文章于 2021-03-04 14:24:44 发布

阅读量186

点赞数

分类专栏：图论 DP

本文链接：https://blog.csdn.net/neweryyy/article/details/111156043

版权

图论同时被 2 个专栏收录

205 篇文章 4 订阅

订阅专栏

159 篇文章 1 订阅

订阅专栏

题意

传送门 POJ 3613 Cow Relays

题解

设从 $u$ 出发，到 $v$ 的长度为 $k$ 的最短路径为 $G_k[u][v]$ ，修改 $F l o y d$ 算法，则有
$G_{k_1+k_2}[u][v]=\min\limits_{1\leq w\leq V}(G_{k_1}[u][w]+G_{k_2}[w][v])$ 用矩阵快速幂求解。初始化矩阵时， $G_1$ 为原图的邻接矩阵，需要注意的是，按照定义，无自环的情况下 $G_1[u][u]=inf$ ；同理 $G_0[u][u]=0$ 。复杂度 $O(T^3\log N)$ 。

#include <algorithm>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
#define maxt 105
#define maxv 1005
#define inf 0x3f3f3f3f
typedef vector<int> vec;
typedef vector<vec> mat;
int N, T, S, E, A[maxt], B[maxt], L[maxt], nv, cp[maxv];

mat mul(mat &C, mat &D)
{
    mat R(C.size(), vec(D[0].size(), inf));
    for (int i = 0; i < C.size(); ++i)
        for (int k = 0; k < D.size(); ++k)
            for (int j = 0; j < D[0].size(); ++j)
                R[i][j] = min(R[i][j], C[i][k] + D[k][j]);
    return R;
}

mat pow(mat &M, int n)
{
    mat R(M.size(), vec(M.size(), inf));
    for (int i = 0; i < M.size(); ++i)
        R[i][i] = 0;
    while (n)
    {
        if (n & 1)
            R = mul(R, M);
        M = mul(M, M);
        n >>= 1;
    }
    return R;
}

int main()
{
    scanf("%d%d%d%d", &N, &T, &S, &E);
    memset(cp, -1, sizeof(cp));
    for (int i = 0; i < T; ++i)
    {
        scanf("%d%d%d", L + i, A + i, B + i);
        cp[A[i]] = cp[A[i]] != -1 ? cp[A[i]] : nv++;
        cp[B[i]] = cp[B[i]] != -1 ? cp[B[i]] : nv++;
    }
    mat M(nv, vec(nv, inf));
    for (int i = 0; i < T; ++i)
    {
        int x = cp[A[i]], y = cp[B[i]];
        M[x][y] = M[y][x] = L[i];
    }
    M = pow(M, N);
    printf("%d\n", M[cp[S]][cp[E]]);
    return 0;
}