HDU 2544 最短路
Problem Description
Input
輸入保證至少存在1條商店到賽場的路線。
Output
Sample Input
Sample Output
Source
Recommend
Solution
三種解法(Floyed、Dijkstra、SPFA)求最短路
Code
1、Floyed:O(n^3)
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stack>
#include <map>
#include <vector>
#include <queue>
#define L 110
#define inf 1000000009
#define LL long long
using namespace std;
int n, m, d[L][L], a, b, c, dis[L];
bool vis[L];
inline void Floyed() {
for (int k = 1; k <= n; ++k)
for (int i = 1; i <= n; ++i)
if (d[i][k] != inf)
for (int j = 1; j <= n; ++j)
if (d[i][j] > d[i][k] + d[j][k]) d[i][j] = d[j][i] = d[i][k] + d[j][k];
printf("%d\n", d[1][n]);
}
int main(){
freopen("HDU2544.in", "r", stdin);
freopen("HDU2544.out", "w", stdout);
while (scanf("%d %d", &n, &m) != EOF && n && m) {
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= n; ++j) d[i][j] = inf;
for (int i = 1; i <= m; ++i)
scanf("%d %d %d", &a, &b, &c), d[a][b] = d[b][a] = c;
Floyed();
}
return 0;
}
2、Dijkstra:O(n^2)
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stack>
#include <map>
#include <vector>
#include <queue>
#define L 110
#define inf 1000000009
#define LL long long
using namespace std;
int n, m, d[L][L], a, b, c, dis[L];
bool vis[L];
inline void dijkstra() {
for (int i = 1; i <= n; ++i) dis[i] = d[1][i], vis[i] = false;
vis[1] = true;
for (int i = 1; i <= n; ++i) {
int temp, minx = inf;
for (int j = 1; j <= n; ++j)
if (!vis[j] && dis[j] < minx) minx = dis[j], temp = j;
vis[temp] = true;
for (int j = 1; j <= n; ++j)
if (!vis[j] && d[j][temp] + dis[temp] < dis[j]) dis[j] = d[j][temp] + dis[temp];
}
printf("%d\n", dis[n]);
}
int main(){
freopen("HDU2544.in", "r", stdin);
freopen("HDU2544.out", "w", stdout);
while (scanf("%d %d", &n, &m) != EOF && n && m) {
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= n; ++j) d[i][j] = inf;
for (int i = 1; i <= m; ++i)
scanf("%d %d %d", &a, &b, &c), d[a][b] = d[b][a] = c;
dijkstra();
}
return 0;
}
3、SPFA:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stack>
#include <map>
#include <vector>
#include <queue>
#define L 10010
#define inf 1000000009
#define LL long long
using namespace std;
struct node {
int nxt, to, w;
} e[L << 1];
int n, m, d[L], head[L], a, b, c, cnt;
bool vis[L];
queue <int> q;
inline void add(int a, int b, int c) {
e[++cnt].nxt = head[a], e[cnt].to = b, e[cnt].w = c, head[a] = cnt;
}
int main(){
freopen("HDU2544.in", "r", stdin);
freopen("HDU2544.out", "w", stdout);
while (scanf("%d %d", &n, &m) != EOF && n && m) {
memset(head, 0, sizeof(head));
memset(vis, 0, sizeof(vis));
for (int i = 2; i <= n; ++i) d[i] = inf;
d[1] = 0, cnt = 0;
for (int i = 1; i <= m; ++i)
scanf("%d %d %d", &a, &b, &c), add(a, b, c), add(b, a, c);
while(!q.empty()) q.pop();
q.push(1);
while (!q.empty()) {
int x = q.front();
q.pop();
vis[x] = 0;
for (int i = head[x]; i; i = e[i].nxt) {
int y = e[i].to;
if (d[y] > d[x] + e[i].w) {
d[y] = d[x] + e[i].w;
if (!vis[y]) vis[y] = 1, q.push(y);
}
}
}
printf("%d\n", d[n]);
}
return 0;
}
Summary
多組數據需要注意預處理