题目:
http://poj.org/problem?id=3613
题意:
一个无向图上有n个点,每条边都有权值,问从点s到点e的恰好经过t条边的最短路径是多少
思路:
利用矩阵快速幂的思想,再套用floyd算法,实际上就是做了t次floyd,利用快速幂加速。其中只有ans结构体对角线置0而其他结构对角线不置0的原因是考虑s和e相同时的情况。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1000 + 10, INF = 0x3f3f3f3f;
int num, has[N];
struct matrix
{
int row, col;
int mat[110][110];
matrix(): row(100), col(100) {}
void init()
{
memset(mat, 0x3f, sizeof mat);
}
};
matrix ans, mp, ta, tb;
void floyd(matrix &a, matrix &b, matrix &c, int f)
{
for(int k = 1; k <= num; k++)
for(int i = 1; i <= num; i++)
for(int j = 1; j <= num; j++)
a.mat[i][j] = min(a.mat[i][j], b.mat[i][k] + c.mat[k][j]);
}
void work(int n)
{
ans.init(), ta.init(), tb.init();
for(int i = 1; i <= num; i++) ans.mat[i][i] = 0;
while(n)
{
if(n & 1)
{
floyd(tb, ans, mp, 1);
ans = tb; tb.init();
}
floyd(ta, mp, mp, 0);
mp = ta; ta.init();
n >>= 1;
}
}
int main()
{
int n, t, s, e;
scanf("%d%d%d%d", &n, &t, &s, &e);
num = 0;
memset(has, 0, sizeof has);
mp.init();
int a, b, c;
for(int i = 1; i <= t; i++)
{
scanf("%d%d%d", &c, &a, &b);
if(! has[a]) has[a] = ++num;
if(! has[b]) has[b] = ++num;
mp.mat[has[a]][has[b]] = mp.mat[has[b]][has[a]] = min(mp.mat[has[a]][has[b]], c);
}
work(n);
printf("%d\n", ans.mat[has[s]][has[e]]);
return 0;
}