最短路之dijkstra算法

上文书咱们提到上次的博客讲了邻接矩阵与邻接表，想学最短路的同学需要要先把他们弄清再继续学习最短路中的各种dfs神奇算法。

dijkstra算法其实可以说是一种贪心，是求单源最短路径的好方法（dijkstra不能处理负边权值）。

洛谷上有板子题，第一道不需要堆优化，第二道需要堆优化。（坑的是第一道中会给同一条边反复赋予值）

以下会用三种方法（1.邻接矩阵；2,邻接表；3，邻接表+堆优化）

邻接矩阵算法思路：

dijkstra只算单源最短路径，也就是在图中某一个点到其他各个点的最短路径，假设起点编号为s；开dis数组，用dis[i]存放当前s到其他节点的最短路径，初值为int类最大整数（2147483647），但显然dis[s]=0；开一个v数组，v[i]=1表示节点i已被当做起点更新过（后面会讲到），反之v[i]=0表示没有；用二维数组map，map[i][j]表示i点到j点的距离，初始化为2147483647，map[i][i]表示自己到自己的距离，即为0。

第一次我们用起点s来更新其他节点，只要满足dis[i]>dis[s]+map[s][i]，那么就用dis[s]+map[s][i]更新dis[i]（然而，此次我们的更新只要是从起点出发的边到达的点都能被更新）。

现在我们想想，现在除了起点外dis值最小的那个点x，从其他某点i（起点除外）到点x必定还要在dis[i]加上从i到x的距离，因为没有负权边，所以i到x的距离必定大于0，本来起点到x的距离就最短，从其他距离大的点到x必定还要加上更大的距离，一定大于现在的dis[x]，因此，现在的dis[x]已经是s到x的最小距离了。

因此我们进行第二步操作，即以x为起点，再更新剩余节点，具体操作见第一次。

.。

.。

.。

.。

最后，所有点都被当做起点更新过了，所以所有点都找到了自己的最短距离，可以输出了。

代码如下：

#include<stdio.h>
#include<string.h> 
/*struct royal{
	int v,w,lo;
}a[500100];
void add(int x,int y,int z)
{
	a[++tot].w=z;
	a[tot].v=y;
	a[tot].lo=last[x];
	last[x]=tot;
}*/
int map[1010][1010],dis[100100],v[100100],n,m,s;
int minx(int a,int b)
{
	if(a<b)return a;
	return b;
}
void dijkstra(int s)
{
	memset(dis,0x3f3f3f,sizeof(dis));
	v[s]=1;
	int minn=2147483647,u;
	for(int i=1;i<=n;i++)
	{
	 dis[i]=map[s][i];
	 if(dis[i]<minn&&i!=s)
	 {
	  minn=dis[i];
	  u=i;
	 }
	}
	for(int i=1;i<n;i++)
	{
	 int f=u,min=0x3f3f3f;
	 v[u]=1;
	 for(int j=1;j<=n;j++)
	  if(!v[j]&&dis[u]+map[u][j]<dis[j])
	   dis[j]=dis[u]+map[u][j];
	 for(int j=1;j<=n;j++)
	  if(!v[j]&&min>dis[j])
	  {
	   min=dis[j];
	   u=j;
	  }
	}
}
int main()
{
	scanf("%d%d%d",&n,&m,&s);
	memset(map,0x3f3f3f,sizeof(map));
	for(int i=1;i<=m;i++)
	{
	 int x,y,z;
	 scanf("%d%d%d",&x,&y,&z);
	 //add(x,y,z);
	 if(map[x][y]>z)
	 map[x][y]=z;
	}
	for(int i=1;i<=n;i++)
	 map[i][i]=0;
	dijkstra(s);
	for(int i=1;i<=n;i++)
	 if(v[i])printf("%d ",dis[i]);
	 else printf("2147483647 ");
	return 0;
}

邻接表算法与邻接矩阵思路一样，只不过调用时改用邻接表。

代码如下（这个可以过第一题）：

#include<stdio.h>
#include<string.h> 
struct royal{
	int v,w,lo;
}a[500100];
int tot=0,last[10100]; 
void add(int x,int y,int z)
{
	a[++tot].w=z;
	a[tot].v=y;
	a[tot].lo=last[x];
	last[x]=tot;
}
int dis[10100],v[10100],n,m,s;
int minx(int a,int b)
{
	if(a<b)return a;
	return b;
}
void dijkstra(int s)
{
	for(int i=1;i<=n;i++)dis[i]=2147483647;
	dis[s]=0;
	long long minn;
	int u=s;
	while(!v[u])
	{
	 minn=2147483647;
	 v[u]=1;
	 for(int j=last[u];j;j=a[j].lo)
	  if(v[a[j].v]==0&&dis[a[j].v]>a[j].w+dis[u])
	  {
	   dis[a[j].v]=a[j].w+dis[u];
	  }
	 for(int j=1;j<=n;j++)
	 if(!v[j]&&minn>dis[j]){
	  minn=dis[j];
	  u=j;
	 }
	}
}
int main()
{
	scanf("%d%d%d",&n,&m,&s);
	for(int i=1;i<=m;i++)
	{
	 int x,y,z;
	 scanf("%d%d%d",&x,&y,&z);
	 add(x,y,z);
	}
	dijkstra(s);
	for(int i=1;i<=n;i++)
	 printf("%d ",dis[i]);
	return 0;
}

堆优化思路：

开一个priority_queue,这是一个优先队列，队首是队列中最大的元素，再定义一个pair，表示优先队列中的元素不是一个数，而是一个二元组，排序会将整个二元组按照二元组里第一个元素进行排序，排序会将编号随值一起排序，而这个排序的速率很快，因此可以省下我们每次查找dis值最小的那个点的时间。因此快乐了很多。我们要找当前最小的点，而默认为队首最大，所以放入负值即可。

代码如下：

#include<stdio.h>
#include<string.h> 
#include<queue>
#include<algorithm>
#include<math.h>
using namespace std; 
struct royal{
	int v,w,lo;
}a[200100];
template<class T>void read(T &x)
{
	x=0;
	int f=0;
	char ch=getchar();
	while(ch<'0'||ch>'9'){f|=(ch=='-');ch=getchar();}
	while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
	x=f?-x:x;
	return;
}
typedef pair<int ,int>dual;
int tot=0,last[100100]; 
void add(int x,int y,int z)
{
	a[++tot].w=z;
	a[tot].v=y;
	a[tot].lo=last[x];
	last[x]=tot;
}
int dis[100100],v[100100],n,m,s;
int minx(int a,int b)
{
	if(a<b)return a;
	return b;
}
void dijkstra(int s)
{
	priority_queue<dual>q; 
	q.push(make_pair(0,s));
	for(int i=1;i<=n;i++)dis[i]=2147483647;
	dis[s]=0;
	while(!q.empty())
	{
	 int x=q.top().second;
	 q.pop();
	 if(v[x])continue;
	 v[x]=1;
	 for(int i=last[x];i;i=a[i].lo)
	  if(dis[a[i].v]>dis[x]+a[i].w)
	  {
	   dis[a[i].v]=dis[x]+a[i].w;
	   q.push(make_pair(-dis[a[i].v],a[i].v));
	  }
	}
}
int main()
{
	read(n),read(m),read(s);
	for(int i=1;i<=m;i++)
	{
	 int x,y,z;
	 read(x),read(y),read(z);
	 add(x,y,z);
	}
	dijkstra(s);
	for(int i=1;i<=n;i++)
	 printf("%d ",dis[i]);
	return 0;
}