优先队列优化的dijsktra的模板

最新推荐文章于 2019-01-30 21:38:03 发布

xinge008

最新推荐文章于 2019-01-30 21:38:03 发布

阅读量529

点赞数

CC 4.0 BY-SA版权

分类专栏：优先队列文章标签：优化 struct

本文链接：https://blog.youkuaiyun.com/xinge008/article/details/7926968

优先队列专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用STL优先队列优化Dijkstra算法的方法，通过定义边结构体并重载比较运算符，实现高效寻找最短路径。文章详细展示了初始化、添加边及执行Dijkstra算法的具体步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用STL的优先队列实现了dijsktra的优化

#define inf 999999999

using namespace std;

const int maxn=1000010;
int dis[maxn];
int head[maxn];
int pnt[maxn];
int next[maxn];
int cost[maxn];
int vis[maxn];
int node,edgenum,edg;

//int start[maxn],end[maxn],wight[maxn];
struct edge
{
	int v,w;
	edge(int vv=0,int ww=0): v(vv),w(ww){}
	bool operator < (const edge& r) const { return w>r.w; }
};

void init()
{
	edgenum=0;
	for(int i=0;i<maxn;i++)
	{
		head[i]=-1;
		vis[i]=0;
		dis[i]=inf;
	}
}

void addedge(int u,int v,int w)
{
	pnt[edgenum]=v;
	cost[edgenum]=w;
	next[edgenum]=head[u];
	head[u]=edgenum++;
}

void dij(int start)
{
	edge mv;
	priority_queue<edge> que;
	int pre=start;
	vis[start]=1;
	dis[start]=0;
	que.push(edge(start,0));
	for(int i=1;i<node;i++)
	{
		for(int j=head[pre];j!=-1;j=next[j])
		{
			int k=pnt[j];
			if(vis[k]==0&&dis[k]>dis[pre]+cost[j])
			{
				dis[k]=dis[pre]+cost[j];
				que.push(edge(k,dis[k]));

			}
		}
		while(!que.empty()&&vis[que.top().v]==1)
			que.pop();
		if(que.empty())
			break;
		mv=que.top();
		que.pop();
		pre=mv.v;
		vis[pre]=1;
	}


}