题意:一个无向图,起点终点已知,求恰好走N步的最短路径。
利用矩阵的变形 a[i][k] *a[k][j] 是不是很像i->j的路径选择,就是利用矩阵乘法的变形,走N步就是要矩阵的N次方。
这个题n<=1000不是很大,但是矩阵解决时三重循环,肯定爆了,所以要离散化。至于为什么离散化成150 我也不知道,反正100是RE,150就过了。
//题意:一个图,起点和终点固定 问经过N步后的最短路径
// 利用矩阵相乘的变形
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define inf 1000000100
#define maxn 150
#define N 300
int ans[maxn][maxn];
int p[1005];
int n,m;
void matrimult(int a[maxn][maxn],int b[maxn][maxn])
{
int i,j,k;
int c[maxn][maxn];
for(i=1;i<=m;i++)
{
for(j=1;j<=m;j++)
{
c[i][j]=inf;
for(k=1;k<=m;k++)
{
c[i][j]=min(c[i][j],a[i][k]+b[k][j]);
}
}
}
for(i=1;i<=m;i++)
for(j=1;j<=m;j++)a[i][j]=c[i][j];
}
void mult(int a[maxn][maxn],int k)
{
int i,j;
for(i=1;i<=m;i++)
for(j=1;j<=m;j++)ans[i][j]=a[i][j];
k--;
while(k>0)
{
if(k&1)
matrimult(ans,a);
k>>=1;
matrimult(a,a);
}
}
int main()
{
// freopen("q.in","r",stdin);
int t,s,e;
int i,j,k;
int a,b,len;
int mat[maxn][maxn];
scanf("%d%d%d%d",&n,&t,&s,&e);
for(i=0;i<maxn;i++)
{
for(j=0;j<maxn;j++)
mat[i][j]=inf;
}
memset(p,-1,sizeof(p));
m=1;
for(i=0;i<t;i++)
{
scanf("%d%d%d",&len,&a,&b);
if(p[a]<0)p[a]=m++;
if(p[b]<0)p[b]=m++;
if(len<mat[p[a]][p[b]])
mat[p[a]][p[b]]=mat[p[b]][p[a]]=len;
}
s=p[s];e=p[e];
mult(mat,n);
cout<<ans[s][e]<<endl;
}