2019牛客暑期多校训练营（第四场）J-free

最新推荐文章于 2019-08-03 21:33:00 发布

原创最新推荐文章于 2019-08-03 21:33:00 发布 · 348 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#最短路 #k层图

图论专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种在图中寻找最短路径的优化算法，特别适用于可以将部分边权重置为0的情况。通过构建k+1层图，利用SPFA算法，实现了在考虑特定边权重变化下的高效路径搜索。

题目链接：https://ac.nowcoder.com/acm/contest/884/J
题意：给你n个顶点的图，能把期中k条边置为0，给定起点，终点，求最短路
题解：把它当成一个k+1层的图，到下一层时，这条边的值为0，dis[i][j]表示第j层，起点到第i个点的距离，可以用spfa

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
#include<vector>
typedef long long ll;
using namespace std;
ll dis[1050][1050], vis[1050][1050];

struct node
{
	ll to, val;
};

vector<node>v[1050];

void spfa(ll s, ll k)
{
	memset(dis, 0x3f3f3f3f3f3f3f3f, sizeof(vis));
	memset(vis, 0, sizeof(vis));
	dis[s][0] = 0;
	vis[s][0] = 1;
	queue<ll>q;
	q.push(s);
	q.push(0);
	while (!q.empty())
	{
		ll u = q.front(); q.pop();
		ll j = q.front(); q.pop();
		for (ll i = 0; i<v[u].size(); i++)
		{
			ll vv = v[u][i].to;
			if (dis[u][j] + v[u][i].val<dis[vv][j])
			{
				dis[vv][j] = dis[u][j] + v[u][i].val;
				if (!vis[vv][j])
				{
					vis[vv][j] = 1;
					q.push(vv);
					q.push(j);
				}
			}
			if (j<k&&dis[u][j]<dis[vv][j + 1])
			{
				dis[vv][j + 1] = dis[u][j];
				if (!vis[vv][j + 1])
				{
					vis[vv][j + 1] = 1; 
					q.push(vv);
					q.push(j + 1);
				}
			}
		}
		vis[u][j] = 0;
	}
}

int main()
{
	ll n, m, s, t, k;
	scanf("%lld %lld %lld %lld %lld", &n, &m, &s, &t, &k);
	for (int i = 1; i <= m; i++)
	{
		ll x, y, z;
		scanf("%lld %lld %lld", &x, &y, &z);
		v[x].push_back({ y,z });
		v[y].push_back({ x,z });
	}
	spfa(s, k);
	ll ansmin = 0x3f3f3f3f3f3f3f3f;
	for (ll i = 0; i <= k; i++)
	{
		ansmin = min(ansmin, dis[t][i]);
	}
	printf("%lld\n", ansmin);
}