HDU2680 Choose the best route【Dijkstra】

Choose the best route

---

题意： 给定一个n个点、m条边的有向图，你有k个起点可以选择，求到终点s的最短路。

题解： 需要注意的只有：

• 多个起点，只要建立一个超级起点，连通这些起点即可。
• 有向图。一开始没看清题意直接按照无向图写wa了一发。
• 两点之间可能多边。用链式前向星存图直接无视这种坑点。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int N = 1e3 + 10;
const int M = 2e4 + 10;
int n, m, s, k;
int dis[N];
int vis[N];
int cnt, head[M];

struct node{
	int id, w;
	bool operator<(const node &A)const{
		return A.w < w;}
};
struct edge{	int to, w, next;} e[M];
void add(int u, int v, int w) {	e[cnt].to = v, e[cnt].w = w, e[cnt].next = head[u], head[u] = cnt++;}
void addedge(int u, int v, int w) {	add(u, v, w);	add(v, u, w);}
void init() {
	cnt = 0;
	for(int i = 0; i < N; i++)	vis[i] = 0, dis[i] = inf;
	for(int i = 0; i < M; i++)	head[i] = -1;
}

void dijkstra() {
	priority_queue<node> Q;
	Q.push({0, 0});	dis[0] = 0;
	while(!Q.empty()) {
		node u = Q.top(); Q.pop();
		if(u.id == s)	break;
		if(vis[u.id])	continue;
		vis[u.id] = 1;
		for(int i = head[u.id]; ~i; i = e[i].next) {
			int v = e[i].to, w = e[i].w;
			if(vis[v])	continue;
			if(dis[v] > u.w + w) {
				dis[v] = u.w + w;
				Q.push({v, dis[v]});
			}
		}
	}
	if(dis[s] == inf)	printf("-1\n");
	else printf("%d\n", dis[s]);
}

int main() {
	while(~scanf("%d%d%d", &n, &m, &s)) {
		init();
		while(m--) {
			int u, v, w;
			scanf("%d%d%d", &u, &v, &w);
			add(u, v, w);
		}
		scanf("%d", &k);
		while(k--) {
			int x;
			scanf("%d", &x);
			add(0, x, 0);
		}
		dijkstra();
	}
	return 0;
}