poj 3613 矩阵解决最短路

题意：一个无向图，起点终点已知，求恰好走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;
	
}