poj 3613(最短路)

题意：求解经过不多于某边数的最短路

思路：矩阵连乘，乘的次数就是不多于某边数的最短路，题目给出的顶点需要映射处理

#include<iostream>
#include<map>
#include<stdio.h>
#include<string.h>
using namespace std;
#define N 202
struct matrix{
    int f[N][N];
    matrix(){
        memset(f,0x3f,sizeof(f));
    }
};
int num;
matrix mul(matrix a,matrix b)
{
    matrix c;
    for(int i=1;i<=num;i++)
    for(int j=1;j<=num;j++)
    for(int k=1;k<=num;k++)
    {
        c.f[i][j]=min(c.f[i][j],a.f[i][k]+b.f[k][j]);
    }
    return c;
}
matrix pow(matrix m,int n)
{
    if(n==1)return m;
    matrix tmp=pow(m,n/2);
    if(n&1)return mul(mul(tmp,tmp),m);
    return mul(tmp,tmp);
}
map<int,int> mp;
int main()
{
    int n,t,s,e;
    num=0;
    scanf("%d%d%d%d",&n,&t,&s,&e);
    mp.clear();
    matrix c;
    for(int i=1;i<=t;i++)
    {
        int a,b,w;
        scanf("%d%d%d",&w,&a,&b);
        if(mp[a]==0)mp[a]=++num;
        if(mp[b]==0)mp[b]=++num;
        c.f[mp[a]][mp[b]]=c.f[mp[b]][mp[a]]=w;
    }
    pow(c,1);
    matrix out=pow(c,n);
    cout<<out.f[mp[s]][mp[e]]<<endl;
    return 0;
}