畅通工程续
Time Limit:1000MS Memory Limit:32768KB 64bit IO Format:%I64d & %I64u
Description
某省自从实行了很多年的畅通工程计划后,终于修建了很多路。不过路多了也不好,每次要从一个城镇到另一个城镇时,都有许多种道路方案可以选择,而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。
现在,已知起点和终点,请你计算出要从起点到终点,最短需要行走多少距离。
Input
本题目包含多组数据,请处理到文件结束。
每组数据第一行包含两个正整数N和M (0
#include "iostream"
#include "cstdio"
#define INF 20000
using namespace std;
int m,n;
struct edge
{
int start,end,weight;
} e[10005];
int dis[105];
void BellmanFord()
{
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
{
if(dis[e[j].start]>dis[e[j].end]+e[j].weight)
dis[e[j].start]=dis[e[j].end]+e[j].weight;
if(dis[e[j].end]>dis[e[j].start]+e[j].weight)
dis[e[j].end]=dis[e[j].start]+e[j].weight;
}
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=0;i<=104;i++)
dis[i]=INF;
for(int i=0;i<m;i++)
{
int u,v,w;
cin>>u>>v>>w;
e[i].start=u;
e[i].end=v;
e[i].weight=w;
}
int s,t;
cin>>s>>t;
dis[s]=0;
BellmanFord();
if(dis[t]==INF)
{
cout<<"-1"<<endl;
}
else
cout<<dis[t]<<endl;
}
}
在Bellman-Ford的基础上的SPFA
一种是我的小伙伴写的,他的博客:http://www.hsuer.com
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int MAXN = 10000 + 5;
const int INF = 0x3f3f3f3f;
struct edgeType {
int v, w, next; //next是下条边的编号
}edge[MAXN];
queue<int> q;
int head[MAXN];
int dis[MAXN], inque[MAXN];//inque记录节点是否已经在队列中
int top, t, n;
void spfa() {
memset(dis, INF, sizeof dis);
memset(inque, 0, sizeof inque);
dis[n] = 0;
q.push(n);
inque[n] = true;
while(!q.empty()) {
int u = q.front();
q.pop();
inque[u] = false;
for(int i = head[u]; i != -1; i = edge[i].next) {
int v = edge[i].v;
if(dis[v] > dis[u] + edge[i].w) {
dis[v] = dis[u] + edge[i].w;
if(!inque[v]) {
inque[v] = true;
q.push(v);
}
}
}
}
}
void add(int u, int v, int w) {
edge[top].v = v;
edge[top].w = w;
edge[top].next = head[u];
head[u] = top++; // 给边编号
}
int main() {
memset(head, -1, sizeof head);
while(!q.empty()) {
q.pop();
}
scanf("%d %d", &t, &n);
top = 0;//top用来给边标号
int u, v, w;
while(t--) {
scanf("%d %d %d", &u, &v, &w);
add(u, v, w);
add(v, u, w);
}
spfa();
printf("%d\n", dis[1]);
return 0;
}
我还看了另一种spfa的写法,这种感觉比较简洁。与上面那种的区别是数据存储方式不一样,代码来源:http://m.blog.csdn.net/blog/u013885699/21110639
#include <iostream>
#include <queue>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAX = 210;
const int INF = 20000;
int map[MAX][MAX];
int dis[MAX];
bool used[MAX];
int SPFA(int start,int end,int n)
{
int i;
for(i=0;i<n;i++)
{
dis[i] = INF;
}
dis[start] = 0;
queue <int> q;
q.push(start);
used[start] = 1;
while(!q.empty())
{
int mid;
mid = q.front();
q.pop();
used[mid] = 0;
for(i=0;i<n;i++)
{
if(map[mid][i] + dis[mid]< dis[i])
{
dis[i] = map[mid][i] + dis[mid];
if(!used[i])
{
q.push(i);
used[i] = 1;//防止下一次此时的i再次存进队列中
}
}
}
}
return dis[end];
}
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
int i,j;
for(i=0;i<MAX;i++)
{
for(j=0;j<MAX;j++)
{
map[i][j] = INF;
}
}//初始化map的值
memset(used,0,sizeof(used));//used数组代表是否访问过
for(i=0;i<m;i++)
{
int a,b,x;
scanf("%d%d%d",&a,&b,&x);
if(x < map[a][b])
{
map[a][b]=x;
map[b][a]=x;
}
}
int s,t;
scanf("%d%d",&s,&t);
int x = SPFA(s,t,n);//储存结果
if(x<INF)
printf("%d\n",x);
else printf("-1\n");
}
return 0;
}