备战2023蓝桥国赛-重新理解dijkstra
前言
第一次学的时候是疫情在家学的,没有弄清楚这个算法的原理,所以这次再次学习的时候会彻底吃透dijkstra算法,后面的bellman-ford,spfa,Floyd等算法之后我也会重新理解一遍,彻底掌握它们。
朴素版dijkstra
题目描述:
解析:朴素版dijkstra是遍历点的个数,复杂度是O(n*n)。
它的大致思路是:
我们先定义数组dist[N],那么dist[i]代表从1到i的最短距离。然后我们找未被找过的dist最小的中继节点,根据这个中继节点来更新其它节点的值,之后重新执行上述操作,通过不断找不同的dist最小的中继节点来更新其它节点的值,循环遍历n次后就能得出1到所有节点的最小值。
首先,解释下什么是中继节点:
看这张图片,从1号点到4号点的路线有1-4、1-2-4、1-3-4,那么1,2,3就是1号点到4号点的中继节点,对,没错,自己也可以是中继节点。
其次,解释下为什么要找dist最小的中继节点:
继续拿上图举例,假设1-2边的值为7,1-3边的值为3,1-4边的值为6,你会拿2号点来更新4还是3号点来更新4呢?毫无疑问是3号,只有3号点才有可能使得1-4路径最小。
最后,解释下为什么这种方法可以解决这个问题:
每次找到dist最小的中继节点时,也可以确定当前dist的最小值,也就是说每遍历一次都能确定至少一个dist的最小值,遍历n次就一定能确定出所有dist的最小值。
为什么呢?
继续拿上图举例,我们知道3号点就是我们确定的最小的中继节点,也确定了dist[3]的最小值就是3,假设dist[3]的最小值不是3,有比它更小的数,那么在寻找dist最小的中继节点时就不应该找到3,而是那个更小的数,这就与前面的矛盾了,所以3就是dist[3]的最小值,证明完毕。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N=510;
int dist[N],g[N][N];
bool st[N];
int n,m;
void dijkstra()
{
memset(dist,0x3f,sizeof(dist));
dist[1]=0;
for(int i=0;i<n;i++)
{
int t=-1;
for(int j=1;j<=n;j++)
{
if(!st[j]&&(t==-1||dist[t]>dist[j]))
t=j;
}
st[t]=true;
for(int j=1;j<=n;j++)
{
dist[j]=min(dist[j],dist[t]+g[t][j]);
}
}
}
int main()
{
scanf("%d%d",&n,&m);
memset(g,0x3f,sizeof(g));
for(int i=1;i<=m;i++)
{
int u,v,c;
scanf("%d%d%d",&u,&v,&c);
g[u][v]=min(g[u][v],c);
}
dijkstra();
if(dist[n]==0x3f3f3f3f) printf("-1");
else printf("%d",dist[n]);
return 0;
}
堆优化版dijkstra
题目描述:
解析:这道题就不能用朴素版的了,因为n太大了,会超时,就得用堆优化版的了。
堆优化就是用到了单链表来存储边,优先队列来存储dist的值和点。它也是找dist最小的中继节点,但找到之后在枚举时是用边来枚举,这样就能使复杂度变成O(mlogn)了(小根堆排序会产生logn的复杂度,所以是mlogn的复杂度)。对于模拟单链表可以到网上找找原理,这里就不赘述了。总的来说弄懂了朴素版的再弄懂堆优化版的就容易多了。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10,M=2e5+10;//N代表点的个数,M代表边的个数
typedef pair<int,int> PII;
int e[M],ne[M],w[M],h[N],n,m,dist[N],idx=0;
bool st[N];
void add(int a,int b,int c)
{
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
void dijkstra()
{
memset(dist,0x3f,sizeof(dist));
dist[1]=0;
priority_queue<PII,vector<PII>,greater<PII> > p;
p.push({0,1});
while(p.size())
{
PII t=p.top();
p.pop();
int ver=t.second,distance=t.first;
if(st[ver]) continue;
st[ver]=true;
for(int i=h[ver];i!=-1;i=ne[i])
{
int j=e[i];
if(dist[j]>dist[ver]+w[i])
{
dist[j]=dist[ver]+w[i];
p.push({dist[j],j});
}
}
}
}
int main()
{
scanf("%d%d",&n,&m);
memset(h,-1,sizeof(h));
while(m--)
{
int u,v,c;
scanf("%d%d%d",&u,&v,&c);
add(u,v,c);
}
dijkstra();
if(dist[n]==0x3f3f3f3f) printf("-1");
else printf("%d",dist[n]);
return 0;
}