Dijkstra算法基本实现
题目要求:实现单源最短路经的迪杰斯特拉算法(Dijkstra),要求:分别采用数组、最小堆实现优先级队列
算法具体思想: 1. 设置一个顶点集合S ,从源点s到集合中的顶点的最短路径的权值已确定
2. 算法反复选择具有最短路估计的顶点 u∈V-S ,并将u加入到S 中.
3. 对u的所有出边进行松弛.
松弛算法:
void relax(int i, int j, int dataij, node *a, int **num, int *front)//front数组存放的是形成最短路径时每个节点前边的节点, dataij是i到j的距离,如果当前源点到j的最短距离大于源点到i加上i到j的距离,那么更新源点到j的最短距离
{
if(a[j].data>a[i].data+num[i][j])
{
a[j].data=a[i].data+num[i][j];
front[j]=i; // j节点的前一个节点是i
}
}
利用最小堆实现优先级队列具体代码,利用STL中的priority_queue
#include<iostream>
#include<queue>
#include<vector>
using namespace std;
struct node
{
int data;
int num;
};
void show(int *front, int i) //利用递归将数组倒序输出,输出最短路径
{
if(front[i]==0)
{
cout<<front[i]<<" ";
//return;
}
else
{
show(front, front[i]);
}
cout<<i<<" ";
return;
}
void relax(int i, int j, int dataij, node *a, int **num, int *front)
{
if(a[j].data>a[i].data+num[i][j])
{
a[j].data=a[i].data+num[i][j];
front[j]=i;
}
}
struct cmp
{
bool operator() (node a, node b)
{
return a.data>b.data;
}
};
int main()
{
vector<int> wwh;
priority_queue<node, vector<node> ,cmp > QueueS;
priority_queue<node, vector<node> ,cmp > QueueQ;
int N;
cin>>N;
int s;
cin>>s;
node a[N];
int front[N];
int num[N][N];
for(int i=0;i<N;i++)
{
if(i!=s) a[i].data=INT_MAX;
else a[i].data=0;
a[i].num=i;
}
for(int i=0;i<N;i++)
{
for(int j=0;j<N;j++)
{
if(i!=j) num[i][j]=INT_MAX;
else num[i][j]=0;
}
}
for(int i=0;i<N;i++)
{
front[i]=0;
}
int jojo, dio;
while(cin>>jojo>>dio&&jojo>=0&&dio>=0)
{
int data;
cin>>data;
num[jojo][dio]=data;
if(jojo==0)
{
a[dio].data=data;
}
}
for(int i=0;i<N;i++)
{
QueueQ.push(a[i]);
}
while(!QueueQ.empty())
{
node u=QueueQ.top();
// cout<<u.num<<" ";
wwh.push_back(u.num);
QueueQ.pop();
QueueS.push(u);
int xuhao=u.num;
for(int i=0;i<N;i++)
{
if(num[xuhao][i]!=INT_MAX)
{
int *p[N];
for(int i=0;i<N;i++) p[i]=num[i];
relax(xuhao, i, num[xuhao][i], a, p, front);
}
}
while(!QueueQ.empty()) QueueQ.pop();
for(int i=0;i<N;i++)
{
int flag=1;
for(int j=0;j<wwh.size();j++)
{
if(wwh[j]==i) flag=0;
}
if(flag) QueueQ.push(a[i]);
}
}
for(int i=0;i<N;i++)
{
cout<<a[i].data<<" ";
}
cout<<endl;
for(int i=0;i<N;i++)
{
show(front, i);
cout<<endl;
}
while(1)
{
}
}
利用数组实现优先级队列的具体代码(这里我可能写的严格来说不是优先级队列, 但是具体的功能是能够满足这个题目的—
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct node
{
int data;
int num;
};
void show(int *front, int i)
{
if(front[i]==0)
{
cout<<front[i]<<" ";
//return;
}
else
{
show(front, front[i]);
}
cout<<i<<" ";
return;
}
bool cmp(node a, node b)
{
return a.data<b.data;
}
class queue
{
private:
vector<node> array;
int L;
public:
queue()
{
array.clear();
L=0;
}
void push(node a)
{
array.push_back(a);
L++;
sort(array.begin(), array.begin()+L, cmp);
}
node top()
{
return array[0];
}
void pop()
{
node t=array[0];
array[0]=array[L-1];
array[L-1]=t;
L--;
sort(array.begin(), array.begin()+L, cmp);
}
bool empty()
{
return L==0;
}
};
void relax(int i, int j, int dataij, node *a, int **num, int *front)
{
if(a[j].data>a[i].data+num[i][j])
{
a[j].data=a[i].data+num[i][j];
front[j]=i;
}
}
int main()
{
vector<int> wwh;
queue QueueS;
queue QueueQ;
int N;
cin>>N;
int s;
cin>>s;
node a[N];
int front[N];
int num[N][N];
for(int i=0;i<N;i++)
{
if(i!=s) a[i].data=INT_MAX;
else a[i].data=0;
a[i].num=i;
}
for(int i=0;i<N;i++)
{
for(int j=0;j<N;j++)
{
if(i!=j) num[i][j]=INT_MAX;
else num[i][j]=0;
}
}
for(int i=0;i<N;i++)
{
front[i]=0;
}
int jojo, dio;
while(cin>>jojo>>dio&&jojo>=0&&dio>=0)
{
int data;
cin>>data;
num[jojo][dio]=data;
if(jojo==0)
{
a[dio].data=data;
}
}
for(int i=0;i<N;i++)
{
QueueQ.push(a[i]);
}
while(!QueueQ.empty())
{
node u=QueueQ.top();
wwh.push_back(u.num);
QueueQ.pop();
QueueS.push(u);
int xuhao=u.num;
for(int i=0;i<N;i++)
{
if(num[xuhao][i]!=INT_MAX)
{
int *p[N];
for(int i=0;i<N;i++) p[i]=num[i];
relax(xuhao, i, num[xuhao][i], a, p, front);
}
}
while(!QueueQ.empty()) QueueQ.pop();
for(int i=0;i<N;i++)
{
int flag=1;
for(int j=0;j<wwh.size();j++)
{
if(wwh[j]==i) flag=0;
}
if(flag) QueueQ.push(a[i]);
}
}
for(int i=0;i<N;i++)
{
cout<<a[i].data<<" ";
}
cout<<endl;
for(int i=0;i<N;i++)
{
show(front, i);
cout<<endl;
}
while(1)
{
}
}
算法具体实现图解