王道机试练习——最短路径最短花费

王道机试练习——最短路径最短花费

题目描述

给你 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<stdio.h>
#include<vector>
using namespace std;
struct E {//邻接链表结构体
	int next;
	int c;
	int cost;
};
vector<E>edge[1001];
int Dis[1001];//距离数组
int cost[1001];//花费数组
bool mark[1001];//是否属于集合K数组
int main() {
	int n, m;
	int S, T;//起点终点
	while (scanf("%d%d", &n, &m) != EOF) {
		if (n == 0 && m == 0) break;
		for (int i = 1; i <= n; i++) edge[i].clear();
		while (m--) {
			int a, b, c, cost;
			scanf("%d%d%d%d", &a, &b, &c, &cost);
			E tmp;
			tmp.c = c;
			tmp.cost = cost;
			tmp.next = b;
			edge[a].push_back(tmp);
			tmp.next = a;
			edge[b].push_back(tmp);
		}
		scanf("%d%d", &S, &T);//输入起点终点
		for (int i = 1; i <= n; i++) {
			Dis[i] = -1;
			mark[i] = false;
		}
		Dis[S] = 0;
		mark[S] = true;
		int newP = S;
		for (int i = 1; i < n; i++) {
			for (int j = 0; j < edge[newP].size(); j++) {
				int t = edge[newP][j].next;
				int c = edge[newP][j].c;
				int co = edge[newP][j].cost;
				if (mark[t] == true) continue;
				if (Dis[t] == -1 || Dis[t] > Dis[newP] + c || Dis[t] == Dis[newP] + c && cost[c] > cost[newP] + co) {
					Dis[t] = Dis[newP] + c;
					cost[t] = cost[newP] + co;
				}
			}
			int min = 123123123;
			for (int j = 1; j <= n; j++) {
				if (mark[j] == true) continue;
				if (Dis[j] == -1) continue;
				if (Dis[j] < min) {
					min = Dis[j];
					newP = j;
				}
			}
			mark[newP] = true;
		}
		printf("%d%d\n", Dis[T], cost[T]);
	}
	return 0;
}