POJ3613 Cow Relays 快速幂最短路！！ 神题

题意：给定一个T(2 <= T <= 100)条边的无向图，求S到E恰好经过N(2 <= N <= 1000000)条边的最短路。

一次floyd我们都走了一个k。

我们执行n-1次即可。

这里我们用快速幂进行处理。

注意离散化

#include <cstdio>
#include <cstring>
#include <map>
using namespace std;
int K, n, m, s, t, x, y, z;
map<int,int>mp;
int min(int a,int b)
{
if(a>b)
return b;
else
return a;
}
struct Matrix {
int a[205][205];
Matrix operator * (Matrix &r) {
Matrix c;
memset(c.a, 0x3f, sizeof c.a);
for(int i = 1; i <= n; i++)
for(int j = 1; j <= n; j++)
for(int k = 1; k <= n; k++)
c.a[i][j] = min(c.a[i][j], a[i][k] + r.a[k][j]);
return c;
}
}st, ans;


void power() {
ans = st, K--;
while(K) {
if(K & 1) ans = ans * st;
st = st * st;
K >>= 1;
}
}


int main() {
scanf("%d%d%d%d", &K, &m, &s, &t);
memset(st.a, 0x3f, sizeof st.a);
while(m--) {
scanf("%d%d%d", &z, &x, &y);
if(mp[x]) x = mp[x]; else x = mp[x] = ++n;
if(mp[y]) y = mp[y]; else y = mp[y] = ++n;
st.a[x][y] = st.a[y][x] = z;
}
power();
printf("%d", ans.a[mp[s]][mp[t]]);
return 0;
}