2019年9月 PAT 7-4 Dijkstra Sequence

考点：Dijistra算法的逻辑
注意点：这一次考察的是Dijistra算法的逻辑，即从起点开始每次选出一个最短路径点（每步更新的d[v]后，可能有多个最小的点更新）与所给序列判断，如果符合则yes。而不是以往考察的求最短路径。切勿先入为主。仔细审查题意。

#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;
const int maxn = 200;
const int INF = 100050;
bool inq[maxn] = {false};
int G[maxn][maxn],d[maxn];
int nv,ne,k,v1,v2,dis;
vector<int> dij;
set<int> now;
bool flag = true;
void dijkstra(int st) {
	flag = true;
	now.clear();
	fill(inq,inq+maxn,false);
	fill(d,d+maxn,INF);
	d[st] = 0;
	int u=st;
	for(int i=1; i<=nv; i++) {
		inq[u] = true;
		int tagmin = INF+1;
		for(int v=1; v<=nv; v++) {
			if(inq[v]==false && G[u][v] < INF) {
				if(d[v] > d[u]+G[u][v]) {
					d[v] = d[u]+G[u][v];
				}
			}
		}
		for(int v=1; v<=nv; v++) {
			if(inq[v]==false && d[v] < tagmin) {
				tagmin=d[v];
				now.clear();
				now.insert(v);
			} else if(d[v]==tagmin) {
				now.insert(v);
			}
		}
		if(i!=nv && now.find(dij[i]) == now.end()) {
			flag = false;
			return;
		} else {
			u = dij[i];
		}
	}
}
int main() {
	fill(G[0],G[0]+maxn*maxn,INF);
	cin>>nv>>ne;
	for(int i=1; i<=ne; i++) {
		cin>>v1>>v2>>dis;
		G[v1][v2]=G[v2][v1]=dis;
	}
	dij.resize(nv);
	cin>>k;
	for(int i=1; i<=k; i++) {
		for(int j=0; j<nv; j++) {
			cin>>dij[j];
		}
		dijkstra(dij[0]);
		if(flag) cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	return 0;
}