2021-08-24四期集训第三天-----Dijkstra算法

最短路问题

 

Dijkstra算法

Dijkstra 的整体思路比较清晰 即进行n（n为n的个数）次迭代去确定每个点到起点的最小值 最后输出的终点的即为我们要找的最短路的距离

所以按照这个思路除了存储图外我们还需要存储两个量

dis[n] //用于存储每个点到起点的最短距离
vis[n]   //用于在更新最短距离时 判断当前的点的最短距离是否确定 是否需要更新

每次迭代的过程中我们都先找到当前未确定的最短距离的点中距离最短的点 （至于为什么是这样那么这就涉及到Dijkstra算法的具体数学证明了 有兴趣的同学可以百度一下）

int t=-1;       //将t设置为-1 因为Dijkstra算法适用于不存在负权边的图
for(int j=1;j<=n;j++)
{
    if(!vis[j]&&(t==-1||dis[t]>dis[j])    //该步骤即寻找还未确定最短路的点中路径最短的点
        t=j;
}

通过上述操作当前我们的t代表就是剩余未确定最短路的点中 路径最短的点 而与此同时该点的最短路径也已经确定我们将该点标记

vis[t]=true;

然后用这个去更新其余未确定点的最短距离

for(int j=1;j<=n;j++)
    dis[j]=min(dis[j],dis[t]+g[t][j]);
//这里可能有同学要问j如果从1开始的话 会不会影响之前已经确定的点的最小距离
//但其实是不会 因为按照我们的Dijkstra算法的操作顺序 先确定最短距离的点的距离已经比后确定的要小 所以不会影响
//当然你也可以在循环判断条件里加上if(!vis[i])
//这里j从1开始只是为了代码的简洁

进行n次迭代后最后就可以确定每个点的最短距离 然后再根据题意输出相应的 要求的最短距离

//Dijkstra朴素版
#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N = 510;

int n, m;
int g[N][N];
int dis[N];
bool vis[N];

int dijkstra() {
	memset(dis, 0x3f3f3f3f, sizeof dis);
	dis[1] = 0;

	for (int i = 0; i < n; i++) {
		int t = -1;
		for (int j = 1; j <= n; j++) 
			if (!vis[j] && (t == -1 || dis[j] < dis[t])) {
				t = j;
			}
		vis[t] = true;
		
		for (int j = 1; j <= n; j++) {
			dis[j] = min(dis[j], dis[t] + g[t][j]);
		}
	}
	if (dis[n] == 0x3f3f3f3f)return -1;
	else return dis[n];
}

int main() {
	scanf("%d%d", &n, &m);
	memset(g, 0x3f3f3f3f, sizeof g);

	for (int i = 1; i <= m; i++) {
		int a, b, k;
		scanf("%d %d %d", &a, &b, &k);
		g[a][b] = min(g[a][b], k);
	}

	int t = dijkstra();

	printf("%d\n", t);
	return 0;
}

优化

 

 

 

注意：若要求任意点i到任意个点j的最短距离，只需修改dijkstra方法中的起源位置dist[i] = 0，以及返回为dist[j]

时间复杂度 O(mlogn)O(mlogn)

每次找到最小距离的点沿着边更新其他的点，若dist[j] > distance + w[i]，表示可以更新dist[j]，更新后再把j点和对应的距离放入小根堆中。由于点的个数是n，边的个数是m，在极限情况下（稠密图m=n*(n−1)/2）最多可以更新m回，每一回最多可以更新n个点（严格上是n - 1个点），有m回，因此最多可以把n^2个点放入到小根堆中，因此每一次更新小根堆排序的情况是O(log(n^2))，一共最多m次更新，因此总的时间复杂度上限是

O(mlog((n^2)))=O(2mlogn)=O(mlogn)

代码

//Dijkstra优先队列优化版
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>

using namespace std;

typedef pair<int, int> PII;

const int N = 150010;

int n, m;
int h[N], e[N], ne[N], w[N], idx;
int dis[N];
bool vis[N];

void add(int a, int b, int c) {
	e[idx] = b; w[idx] = c; ne[idx] = h[a]; h[a] = idx++;
}

int dijkstra() {
	memset(dis, 0x3f, sizeof dis);
	dis[1] = 0;
	priority_queue<PII, vector<PII>, greater<PII>>heap;
	heap.push({ 0,1 });

	while (heap.size()) {
		auto t = heap.top();
		heap.pop();

		int ver = t.second, distance = t.first;
		if (vis[ver])continue;
		vis[ver] = true;

		for (int i = h[ver]; i != -1; i = ne[i]) {
			int j = e[i];
			if (dis[j] > distance + w[i]) {
				dis[j] = distance + w[i];
				heap.push({ dis[j],j });
			}
		}
	}

	if (dis[n] == 0x3f3f3f3f)return -1;
	return dis[n];
}

int main() {
	scanf("%d%d", &n, &m);
	memset(h, -1, sizeof h);

	while (m--) {
		int a, b, k;
		scanf("%d %d %d", &a, &b, &k);
		add(a, b, k);
	}

	int t = dijkstra();

	printf("%d\n", t);
	return 0;
}