【PAT】Dijkstra

概述

Dijkstra算法主要用于解决单源最短路径问题。在题目中，一般除了求最短路径还会附加其他条件加以筛选，求得唯一路径。

标准套路

st=起点，en=终点，D[ ]=到各点的最短路径，dis[ ][ ]=各点之间的路径长度（注意，a到b和b到a都要存储），visit[ ]=访问控制，pre[ ]=各点的前驱结点

首先要将不能用0初始化的各对象用fill函数初始化

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

const int inf=99999;
fill(D,D+505,inf);
fill(dis[0],dis[0]+505*505,inf);

接下来就是dij的主要算法

void dij(){
	D[st]=0;
	for(int i=0;i<N;i++){
		int u=-1,dmin=inf;
		for(int j=0;j<N;j++){
			if(visit[j]==0&&D[j]<dmin){
				dmin=D[j];
				u=j;
			}
		}
		if(u==-1) break;
		visit[u]=1;
		for(int v=0;v<N;v++){
			if(visit[v]==0&&dis[u][v]!=inf){
				if(D[u]+dis[u][v]<D[v]){
					D[v]=D[u]+dis[u][v];
					pre[v]=u;
				}
			}
		}
	}
}

最后得到的是所有结点的前驱数组pre[]和最短路径长度数组D[]，如果需要输出路径，则可以利用pre[]从终点往前驱逐个输出。

变化

最短路径通常不会是唯一的，所以题目中一般会根据情景设置一些其他条件，来达到唯一路径的目的。对于这些条件首先需要观察是否满足最优子结构，如果满足，则只需要在更新最短路径时添加筛选条件即可，否则需要用vector<vector<int>> pre来存储不同的路线，最后再使用一次dfs来筛选唯一路径。
另外，如果最后不需要输出路径，且其他条件满足最优子结构不需要dfs时，则可以不用存储pre[]数组。