【PAT甲级】1030 Travel Plan (30 分)（SPFA,DFS）

题意：

输入N,M,S,D（N,M<=500,0<S,D<N）,接下来M行输入一条边的起点，终点，通过时间和通过花费。求花费最小的最短路，输入这条路径包含起点终点，通过时间和通过花费。

trick：

找了半小时bug终于发现是给dis数组赋初值时范围是1~N，这道题点是从0~N-1的，故初次只通过了第0组数据，初始化改一下边界即可AC。

代码：

#define HAVE_STRUCT_TIMESPEC
#include<bits/stdc++.h>
using namespace std;
int n,m,s,d;
int a[507];
typedef struct road{
int x,y,z;
};
vector<road>edg[507];
vector<pair<int,int> >pre[507];
vector<pair<int,int> >tmp_path,path;
int dis[507];
bool vis[507];
int ans=1e9;
void spfa(){
for(int i=0;i<n;++i)
dis[i]=1e9;
dis[s]=0;
vis[s]=1;
queue<int>q;
q.push(s);
while(!q.empty()){
int x=q.front();
vis[x]=0;
q.pop();
for(int i=0;i<edg[x].size();++i){
int v=edg[x][i].x;
int w=edg[x][i].y;
int val=edg[x][i].z;
if(dis[v]>dis[x]+w){
dis[v]=dis[x]+w;
pre[v].clear();
pre[v].push_back({x,a[val]});
if(!vis[v]){
vis[v]=1;
q.push(v);
}
}
else if(dis[v]==dis[x]+w)
pre[v].push_back({x,a[val]});
}
}
}
void dfs(int x,int y){
if(x==s){
if(y<ans){
ans=y;
path=tmp_path;
}
}
tmp_path.push_back({x,y});
for(int i=0;i<pre[x].size();++i)
dfs(pre[x][i].first,y+pre[x][i].second);
tmp_path.pop_back();
}
int main(){
cin>>n>>m>>s>>d;
int u,v,w;
for(int i=1;i<=m;++i){
cin>>u>>v>>w>>a[i];
edg[u].push_back({v,w,i});
edg[v].push_back({u,w,i});
}
spfa();
dfs(d,0);
cout<<s<<" ";
for(int i=path.size()-1;i>=0;--i)
cout<<path[i].first<<" ";
cout<<dis[d]<<" "<<ans;
return 0;
}

转载于:https://www.cnblogs.com/ldudxy/p/11505317.html