CF-Round #636-div3-E题

CF-Round #636-div3-E题

E. Weights Distributing

传送门

这道题是图的问题，bfs~

题目大意：给你一个无向无权重的图，假设这个图有n个结点，m条边。你的小伙伴要从所给的结点a到达结点b，再从结点b到达结点c，每经过一条边需要给相应的费用。这个费用已经给出，m个值。
要求你给这个图的边赋上权值，使得你的小伙伴需要付出的总费用最小

本题思路：要求我们付出的总费用最小，也就是要求我们走过的路最少，走过的边最少，我们当然希望从结点a到结点b，从结点b到结点c直接的边走下来是一条直线。（这样可以使得我们走的路最少了，继而总费用最小）。
对了，前面我们输入的时候把费用sort一遍。
然后需要求一遍前缀和pre[]。
上面只是一种理想情况。
我们需要考虑一般情况。

一般情况：不能构成直线的情况，也就是我们要重复经过走过的边。
我们不妨把结点a到结点b走过的路看成一条直线，结点b到结点c走过的路当成一条直线。如果要重复经过走过的边的话，那么这两条直线就会有交点。这个交点设为x；
那么我们可以把小伙伴走过的路分为：
从a到x, 从x到b，从b到x，从x到c；
那么观察可以发现从x到b的路需要走两次。
那我们不妨贪心的把x到b的路线上的权值设为最小。

现在我们需要记录a, b, c到其他的每个结点最短路径是多少。记录下来。
这个地方我们可以跑bfs；
三遍bfs即可实现。
然后再跑一遍找到最合适的交点x，然后赋值
这里注意一点：我们现在求的是一般情况：从a到x, 从x到b，从b到x，从x到c；
如果找到的这个交点x到达a, b, c的路线长度已经超过了m，代表我们选择的交点x位置不当，这样会让我们额外花费不该花费的费用。

所以我们可以通过bfs得出来得路线的长度，交点x分别到达结点a, b, c得路线长度之和如果大于了m（图中边的数量），我们直接continue即可。

然后我们把b到x的路线长度为t的话，赋值为费用最小的前t个数，则我们需要花费的就是pre[t];

维护最小值即可。

好啦~完毕！

代码部分：

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
const int INF = 1e9;

int n, m, a, b, c;
vector<vector<int> > g;

void bfs(int now, vector<int> &c)
{
	c[now] = 0;
	queue<int> q;
	q.push(now);
	while (!q.empty())
	{
		int v = q.front();
		q.pop();
		int siz = g[v].size();
		for (int i = 0; i < siz; i++)
		{
			if (c[g[v][i]] == INF)
			{
				c[g[v][i]] = c[v] + 1;
				q.push(g[v][i]);
			}
		}
	}
}

int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		cin >> n >> m >> a >> b >> c;
		--a, --b, --c;
		vector<int> p(m);
		for (int i = 0; i < m; i++)
		{
			scanf ("%d", &p[i]);
		}
		sort(p.begin(), p.end());
		vector<ll> pre(m + 1);
		for (int i = 0; i < m; i++)
		{
			pre[i + 1] = pre[i] + p[i];
		}
		g = vector<vector<int> >(n);
		for (int i = 0; i < m; i++)
		{
			int x, y;
			scanf ("%d%d", &x, &y);
			--x;
			--y;
			g[x].push_back(y);
			g[y].push_back(x);
		}
		vector<int> ap(n, INF);
		vector<int> bp(n, INF);
		vector<int> cp(n, INF);
		bfs(a, ap);
		bfs(b, bp);
		bfs(c, cp);
		ll ans = 1e18;
		for (int i = 0; i < n; i++)
		{
			if (ap[i] + bp[i] + cp[i] > m)
			{
				continue;
			}
			ans = min(ans, pre[bp[i]] + pre[ap[i] + bp[i] + cp[i]]);
		}
		cout << ans << endl;
	}
	return 0;
}