Luogu_P2886 [USACO07NOV]牛继电器Cow Relays【题解】图论 矩阵快速幂

一世执手 提交于 2019-11-30 16:21:40

Luogu_P2886 [USACO07NOV]牛继电器Cow Relays

### 图论 矩阵快速幂

题目链接
POJ3613
题意就是求经过k条边的从st到ed的最短路
我们假设$a[i][j]$为经过一条边的最短路
$a2[i][j]$为经过两条边的。
那么$a2[i][j]=min(a[i][k]+a[k][j])$
那么其实这就是一个类似矩阵快速幂的东西
只不过是把$+$和$*$变成$min$和$+$
那么就把原数离散化之后矩阵运算


代码如下:

#include<cstdio>
#include<cstring>
#include<map>
using namespace std;
const int maxn=300,inf=0x3f3f3f3f;
int n,m,st,ed,tot;
map<int,int> ls;
struct node{
    int c[maxn][maxn];
}ans,base;
inline node mul(node x,node y){
    node res;memset(res.c,inf,sizeof(res.c));
    for(int i=1;i<=tot;i++)
        for(int k=1;k<=tot;k++){
            if(x.c[i][k]>=inf) continue;
            for(int j=1;j<=tot;j++)
                res.c[i][j]=min(res.c[i][j],x.c[i][k]+y.c[k][j]);
        }
    return res;
}
inline void qp(int x){
    while(x){
        if(x&1) ans=mul(ans,base);
        base=mul(base,base);
        x>>=1;
    }
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&st,&ed);tot=0;
    memset(base.c,inf,sizeof(base.c));memset(ans.c,inf,sizeof(ans.c));
    for(int i=1;i<=m;i++){
        int x,y,z;scanf("%d%d%d",&z,&x,&y);
        x=ls[x] ? ls[x] : (ls[x]=++tot);
        y=ls[y] ? ls[y] : (ls[y]=++tot);
        ans.c[x][y]=ans.c[y][x]=z;base.c[x][y]=base.c[y][x]=z;
    }
    qp(n-1);
    printf("%d\n",ans.c[ls[st]][ls[ed]]);
    return 0;
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!