上集回顾:最短路:dijkstra算法_物联网土猫的博客-CSDN博客
那么按照写文章的尿素,接下来就是我们的优化时间;
纵观代码,我们可以找到一些地方优化,比如:我们能不能在循环上缩减,减少遍历次数呢?答案当然是可以的!
看看代码,大循环肯定不能减,小循环呢?最后一步判断最短路比较肯定不能删掉,那么我们理所当然的应该改动寻找最小权值那里。
这里的最小权值只是一个值,我们需要的是最小权值这个点,所以这里就有了一种、堆优化方案,具体是,第一次循环我们找到最短路,接下来把出边更新之后,我们把所有出边加入我们的堆,这里我们的堆是priority_queue,他实现了堆内部自动排序,但是有个问题是他是大根堆,也就是从大到小排列,我们这里存入权值的时候存入负值也就直接变成小根堆了。
这样的话我们每次只需要取出第一个值,接着遍历,加入即可;
这里的代码我们定义了自动排序的小跟堆,也就不需要存入负值了;
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N=10010;
int n,m;
int head[N],w[N],e[N],ne[N],idx;
int dist[N];
bool st[N];
typedef pair<int,int> PII;
void add(int a,int b,int c)
{
e[idx]=b,ne[idx]=head[a],w[idx]=c,head[a]=idx++;
}
int dijkstra()
{
memset(dist,0X3f3f3f3f,sizeof(dist));
dist[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, distence=t.first;
if(st[ver]) continue; //如果已经被找过最短路了,就不需要再加入一次,否则会无限循环
for(int i=head[ver];i!=-1;i=ne[i])
{
int j=e[i];
if(dist[j]>distence+w[i])
{
dist[j]=distence+w[i];//常规操作 W
heap.push({dist[j],j});
} //加入操作;
}
}
if(dist[n]==0X3f3f3f3f) return -1;
else return dist[n];
}
int main()
{
memset(head,-1,sizeof(head));
cin>>n>>m;
while(m--)
{
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
}
int t=dijkstra();
cout<<t<<endl;
return 0;
}
具体代码: