P8802 [蓝桥杯 2022 国 B] 出差--最短路

点击跳转例题

思路：
最短路的模板题，利用优先队列来解题即可。dijkstra算法。
多了一个点权值得计算。
 

#include <bits/stdc++.h>
#define int long long //(有超时风险) 
#define PII pair<int,int>
#define endl '\n'
#define ll long long 
#define LL __int128

using namespace std;

const int N=2e6+10,M=2e3+10,mod=998244353,INF=0x3f3f3f3f;

int n,m;
int a[M];
int dist[M];
bool st[M];
int h[N],e[N],ne[N],w[N],idx;

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

void dijkstra()
{
	priority_queue<PII,vector<PII>,greater<PII>>q;
	q.push({0,1});
	
	memset(st,false,sizeof st);
	memset(dist,0x3f,sizeof dist);
	dist[1]=0;
	
	while(q.size())
	{
		PII t=q.top();q.pop();
		if(st[t.second])continue;
		st[t.second]=true;
		
		//边i~j,边权存在w[i]上，然后加上j的隔离时间； 
		for(int i=h[t.second];~i;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>dist[t.second]+w[i]+a[j])
			{
				dist[j]=dist[t.second]+w[i]+a[j];
				q.push({dist[j],j}); 
			}
		}
	}
	//最后减去最后的城市，不用隔离； 
	cout<<dist[n]-a[n]<<endl;
}

signed main()
{
	std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
	
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>a[i];
	
	memset(h,-1,sizeof h);
	while(m--)
	{
		int x,y,z;cin>>x>>y>>z;
		add(x,y,z); 
		//致命错误，没有看到双向边； 
		add(y,x,z);
	}
	dijkstra();
	return 0;
 }