在牛客网上做的一道题,做完以后加了很多注释以防自己忘记。
题目描述
给你n个点,m条无向边,每条边都有长度d和花费p,给你起点s终点t,要求输出起点到终点的最短距离及其花费,如果最短距离有多条路线,则输出花费最少的。
输入描述:
输入n,m,点的编号是1~n,然后是m行,每行4个数 a,b,d,p,表示a和b之间有一条边,且其长度为d,花费为p。最后一行是两个数 s,t;起点s,终点t。n和m为0时输入结束。 (1<n<=1000, 0<m<100000, s != t)
输出描述:
输出 一行有两个数, 最短距离及其花费。
#include<iostream>
#include<queue>
#include<vector>
using namespace std;
struct edge{
int to;
int d;
int cost;
edge(int a,int b,int c):to(a),d(b),cost(c){}
};
struct ver{//顶点节点,在确定了源点以后才定义,是用于Dijkstra算法的过程变量,不用统一为每个顶点建立,随用随建
int d;//该点到源点的距离
int no;//节点编号
ver(int x,int y):no(x),d(y){}
bool operator<(const ver&b)const{//优先队列默认降序排列,所以重载时大于小于号调转
return d>b.d;
}
};
vector<edge> graph[1001];//邻接表存储图
int dis[1001]={0};//存到源点的距离
int cost[1001]={0};
void build(int s){//s是源点编号
dis[s]=0;
cost[s]=0;
priority_queue<ver> q;
q.push(ver(s,0));
while(!q.empty()){//Dijkstra算法算法
int no=q.top().no;//新加入的点的编号
int d=q.top().d;
q.pop();
vector<edge>::iterator it;
for(it=graph[no].begin();it<graph[no].end();it++){//遍历和新加入的点(no)直接相连的点(it->to)
if(dis[no]+it->d<dis[it->to]||(dis[no]+it->d==dis[it->to]&&cost[no]+it->cost<cost[it->to])){
//如果经过当前点到it点的距离变小或者距离相等但花费变小
dis[it->to]=dis[no]+it->d;//更新距离数组和花费数组
cost[it->to]=cost[no]+it->cost;
q.push(ver(it->to,dis[it->to]));//it点入队
}
}
}
}
int main(){
int n,m,d,p,s,t,a,b;
while(cin>>n>>m&&n!=0){
for(int i=0;i<m;i++){//建无向图邻接表
cin>>a>>b>>d>>p;
graph[a].push_back(edge(b,d,p));//注意每条边对应两个边节点
graph[b].push_back(edge(a,d,p));
}
fill(dis,dis+n+1,2000000000);//距离数组和花费数组初始化为最大值
fill(cost,cost+n+1,2000000000);
cin>>s>>t;
build(s);
cout<<dis[t]<<" "<<cost[t]<<endl;
}
}