昨天刚写了一篇Bellman-Ford的文章,还没看的快去看,https://blog.csdn.net/weixin_43417464/article/details/116072823?spm=1001.2014.3001.5502顺便给我刷个三连
昨天提到了图论的另外一个没有地基的千丈大楼:Dijkstra。可是他自己也有自己的优点,那就是:上限高的很(高到你到顶楼都可以出高原反应,然后你就因为氧气不足而死,所以测试这事情还是留给CCF吧)
老套路,先讲思路:
- 先找到起点,把他的最小到达权值定为0
- 把它能到的里面挑一条距离最短的进行疏松,然后找次短的......
当你把每一个点都像2一样操作完了,你就可以输出了。
#include<bits/stdc++.h>
using namespace std;
int a,b,c,m,n;
bool vis[100001];
int d[500001];
struct road{
int from,to,len;
};
vector<road> v[100001];
void Dijkstra(int start,int end){
for(int i=1;i<=n;i++) d[i]=1e9;
d[start]=0;
for(int i=1;i<=n;i++){
int from=1e9;
for(int j=1;j<=n;j++)
if(!vis[j]&&(from==1e9||d[from]>d[j])) from=j;
if(from==1e9) break;
vis[from]=1;
int to,len;
for(int j=0;j<v[from].size();j++){
to=v[from][j].to,len=v[from][j].len;
if(d[from]+len<d[to]) d[to]=d[from]+len;
}
}
printf("%d\n",d[end]);
}
int main(){
cin>>n>>m;
for(int i=0;i<m;i++){
scanf("%d %d %d",&a,&b,&c);
v[a].push_back((road){a,b,c});
v[b].push_back((road){b,a,c});
}
Dijkstra(1,n);
return 0;
}
但是,这样的代码还是不为最完美的,他还是很耗时间。可以算出,他的时间复杂度为O(n*n)
所以,我们就要用到一个神奇的数据结构:优先队列(priority_queue),用这个工具来帮我们排序好(因为时间复杂度大的原因是找距离最短又嵌了一层for)
#include<bits/stdc++.h>
using namespace std;
int a,b,c,m,n;
bool vis[100001];
int d[100001];
struct road{
int from,to,len;
};
struct node{
int id,len;
friend bool operator < (const node &a,const node &b){
return a.len>b.len;
}
};
vector<road> v[100001];
void pqDijkstra(int start,int end){
for(int i=1;i<=n;i++) d[i]=1e9;
d[start]=0;
priority_queue<node> pq;
pq.push((node){start,0});
while(!pq.empty()){
node head=pq.top();pq.pop();
int from=head.id,to,len;
if(vis[from]) continue;
vis[from]=1;
for(int i=0;i<v[from].size();i++){
to=v[from][i].to,len=v[from][i].len;
if(d[from]+len<d[to]){
d[to]=d[from]+len;
pq.push((node){to,d[to]});
}
}
}
printf("%d\n",d[end]);
}
void Dijkstra(int start,int end){
for(int i=1;i<=n;i++) d[i]=1e9;
d[start]=0;
for(int i=1;i<=n;i++){
int from=1e9;
for(int j=1;j<=n;j++)
if(!vis[j]&&(from==1e9||d[from]>d[j])) from=j;
if(from==1e9) break;
vis[from]=1;
int to,len;
for(int j=0;j<v[from].size();j++){
to=v[from][j].to,len=v[from][j].len;
if(d[from]+len<d[to]) d[to]=d[from]+len;
}
}
printf("%d\n",d[end]);
//如果要输出到每一个的话就是for一下,输出d[i]就OK
}
int main(){
cin>>n>>m;
for(int i=0;i<m;i++){
scanf("%d %d %d",&a,&b,&c);
v[a].push_back((road){a,b,c});
v[b].push_back((road){b,a,c});
}
pqDijkstra(1,n);
return 0;
}
好的,然后练一道洛谷的题3371https://www.luogu.com.cn/problem/P3371
这道题可以不用pq,直接常规来,把输入输出格式改一下,交上去就A了,这里直接讲用pq的(其实pq代码交到4779也能过)
#include<bits/stdc++.h>
using namespace std;
int d[100001],n,m,a,b,c,s;bool vis[100001];
struct road{
int from,to,len;
};
struct node{
int id,len;
friend bool operator < (const node &a,const node &b){
return a.len>b.len;
}
};
vector<road> v[100001];
void pqDijkstra(int start){
for(int i=1;i<=n;i++) d[i]=INT_MAX;
d[start]=0;
priority_queue<node> pq;
pq.push((node){start,0});
while(!pq.empty()){
node head=pq.top();pq.pop();
int from=head.id,to,len;
if(vis[from]) continue;
vis[from]=1;
for(int j=0;j<v[from].size();j++){
to=v[from][j].to,len=v[from][j].len;
if(d[from]+len<d[to]){
d[to]=d[from]+len;
pq.push((node){to,d[to]});
}
}
}
for(int i=1;i<=n;i++) printf("%d ",d[i]);
}
int main(){
scanf("%d %d %d",&n,&m,&s);
for(int i=0;i<m;i++){
scanf("%d %d %d",&a,&b,&c);
v[a].push_back((road){a,b,c});
}
pqDijkstra(s);
return 0;
}