天天看点

PAT A 2019年秋季 7-4 Dijkstra Sequence AC代码

一道dijkstra算法的模板题

对于所谓的Dijkstra Sequence,直接判断给出结点序列到起点距离(d[x])是否是不严格增序(<=)即可,这样只需要用模板,再判断距离是否不严格递增即可。

对于这一点,我的理解是:

根据dijkstra算法思想原理,算法每一轮会确定一个离起点最近的点的距离,并把这个点作为这一轮的间接点,所以算法依次取到的间接点,一定是标记为未确定的余下所有点中,到起点距离最近的点(而可能有多个点距离都是最近,所以产生了不同的序列),所以这就保证了取到的间接点序列到起点的距离是不严格递增的。并且只要是不严格递增的,就可以是间接点取到的顺序。

PAT A 2019年秋季 7-4 Dijkstra Sequence AC代码
#include<iostream>
#include<vector>
using namespace std;
int Nv,Ne,K;
int inf=1000000000;
int G[1001][1001];
int d[1001];
void dijkstra(int s){
	fill(d,d+1001,inf);	
	int visit[1001]={0};
	d[s]=0;
	for(int i=1;i<=Nv;i++){
		int min=inf,u=-1;
		for(int j=1;j<=Nv;j++){
			if(visit[j]==0&&d[j]<min){
				min=d[j];
				u=j;
			}
		}
		if(u==-1) return;
		visit[u]=1;
		for(int v=1;v<=Nv;v++){
			if(G[u][v]+d[u]<d[v]&&G[u][v]!=inf&&visit[v]==0){
				d[v]=G[u][v]+d[u];
			}
		}
	}
}
int main(){
	fill(G[0],G[0]+1001*1001,inf);
	cin>>Nv>>Ne;
	for(int i=0;i<Ne;i++){
		int a,b,s;
		scanf("%d %d %d",&a,&b,&s);
		G[a][b]=s;
		G[b][a]=s;
	}
	cin>>K;
	for(int i=0;i<K;i++){
		bool flag=true;
		int start;
		scanf("%d",&start);
		vector<int> v;
		v.push_back(start);
		for(int j=1;j<Nv;j++){
			int index;
			scanf("%d",&index);
			v.push_back(index);
		}
		dijkstra(start);
		for(int j=1;j<v.size();j++){
			if(d[v[j-1]]>d[v[j]]){
				flag=false;
				break;
			}
		}
		if(flag){
			printf("Yes\n");
		}else{
			printf("No\n");
		}
	}
	return 0;
}