Dijkstra求最短路 II (邻接表+dij()堆优化模板）

活动 - AcWing 系统讲解常用算法与数据结构，给出相应代码模板，并会布置、讲解相应的基础算法题目。https://www.acwing.com/problem/content/852/AcWing 850

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<map>
#include<vector>
#include<set>
#include<cmath>
#include<queue>
//#define int long long
#define endl "\n"


using namespace std;
const int N=2e5+10;

int e[N],h[N],w[N],ne[N],idx;
int dis[N],book[N],n,m;

void add(int a,int b,int c)
{
	e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}

void dij()
{
	memset(dis,0x3f,sizeof dis);
	dis[1]=0;
	
	queue<int> q;
	q.push(1);
	book[1]=1;
	
	while(q.size())
	{
		auto t=q.front();
		q.pop();
		book[t]=0;
		
		for(int i=h[t];i!=-1;i=ne[i])
		{
			int j=e[i]; 
			if(dis[j]>dis[t]+w[i])
			{
				dis[j]=dis[t]+w[i];
				if(book[j]==0)
				{
					q.push(j);
					book[j]=1;
				}
			}
		}
	}
	if(dis[n]==0x3f3f3f3f) cout<<"-1";
	else cout<<dis[n];

}

void solve()
{
	memset(h,-1,sizeof h);
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int a,b,c;cin>>a>>b>>c;
		add(a,b,c);
	} 
	
	dij();
} 

signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int T=1;//cin>>T;
	while(T--) solve();
}