dijkstra,多源最短路径
1.原始版(n方)
distance数组、intree数组、parent数组
首先将所有的数组初始化
将始点纳入到树中(到树的距离为0)
并赋intree 为true
遍历该点的所有边,更新其到树的最小距离
搜索所有的点,找到距离树最小的点
将该点纳入树中,开始下一轮循环
直到所有点都被纳入到树中
2.优先队列优化版(对数n)
需要多添加一个数组done来判断所弹出的点是否被处理过
将始点压入队列中,此时最小优先队列有且只有始点
弹出队列顶点
遍历所有边
更新数据,将更新后的数据重新排序(可以只将数据重新插入而不删除,通过done数组来判断是否已经操作)
再次弹出顶点,重复。
直到优先队列为空
struct Edge
{
int from,to,dist
}
struct HeapNode
{
int d,u;//d是距离,u是节点号
bool operator < (const HeapNode& rhs) const
{
return d>rhs.d;
}
};
struct Dijkstra
{
int n,m;//vertices and edges
vector<Edge> edges;
vector<int> G[maxn];
bool done[maxn];
int d[maxn];
int p[maxn];
void init(int n)
{
this->n=n;
for(int i=0;i<n;i++) G[i].clear();
edges.clear();
}
void AddEdge(int from,int to,int dist)
{
edges.push_back((Edge){from,to,dist});
m=edges.size();
G[from].push_back(m-1);
}
void dijkstra(int s)
{
priority_queue<HeapNode> Q;//有限队列优化
for(int i=0;i<n;i++) d[i]=INF;//初始化
d[s]=0;
memset(done,0,sizeof(done));
Q.push((HeapNode){0,s});//把节点压栈开始
while(!Q.empty())//当队列非空
{
HeapNode x=Q.top();//队首元素
Q.pop();//弹出
int u=x.u;
if(done[u]) continue;
done[u]=true;
for(i=0;i<G[u].size();i++)//遍历所有的边
{
Edge& e=edges[G[u][i]];
if(d[e.to]>d[u]+e.dist)
{
d[e.to]=d[u]+e.dist;
p[e.to]=G[u][i];
Q.push((HeapNode){d[e.to],e.to});
}
}
}
}
};