【2022年11月24日提高A组】 普通快乐【最短路】

在这里插入图片描述

思路:

改造版的dij,记录每个点的最短路从哪个奇葩点来,然后每次更新答案

c o d e code code

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#define ll long long

using namespace std;

const ll MAXN = 1e5 + 10;

ll n, m, k, dis[MAXN][2];
ll tot, head[MAXN], ans = 1e18;
struct node {
	ll to, next, w;
}b[MAXN * 5];
bool v[MAXN];
priority_queue<pair<ll, pair<ll, ll> > > q;

void add(ll x, ll y, ll w) {
	b[++ tot] = (node) { y, head[x], w };
	head[x] = tot;
}

void dij() {
	while(!q.empty()) {
		ll len = q.top().first, x = q.top().second.first, fa = q.top().second.second;
		q.pop();
		if(v[x]) continue;
		v[x] = 1;
		dis[x][1] = fa;
		for(ll i = head[x]; i; i = b[i].next) {
			ll y = b[i].to;
			if(dis[y][1] != dis[x][1] && dis[y][0] + dis[x][0] + b[i].w < ans) ans = dis[y][0] + dis[x][0] + b[i].w;
			if(dis[y][0] > dis[x][0] + b[i].w) {
				dis[y][0] = dis[x][0] + b[i].w;
				q.push(make_pair(-dis[y][0], make_pair(y, fa)));
			}
		}
	}
}

int main() {
	scanf("%lld%lld%lld", &n, &m, &k);
	for(ll i = 1; i <= m; i ++) {
		ll x, y, w;
		scanf("%lld%lld%lld", &x, &y, &w);
		add(x, y, w), add(y, x, w);
	}
	memset(dis, 0x7f, sizeof(dis));
	for(ll i = 1; i <= k; i ++) {
		ll x;
		scanf("%lld", &x);
		dis[x][0] = 0;
		dis[x][1] = x;
		q.push(make_pair(0, make_pair(x, x)));
	}
	dij();
	printf("%lld", ans);
	return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值