Codeforces Round #636 (Div. 3) E. Weights Distributing（最短路+贪心）

题目链接


题意：给定一棵树，和一个权值数组，要求你选择一种方案把权值数组赋值到树的每条边上去，最后使得a->b->c的路径和最小，只需输出最小值即可。
思路：要想a->b,b->c最短的话我们很显然要他们能重合的部分尽量多，如此一下就可以尽量取前面小一点的前缀和了。

#include<bits/stdc++.h> 
using namespace std;
const int maxn =2e5+1; 
typedef long long ll;
vector<int>g[maxn];
ll num[maxn],sum[maxn];
bool vis[maxn];
int T,n,m,a,b,c,u,v,d[maxn][3];
void spfa(int x,int id)
{
	memset(vis,false,sizeof(vis));
	for(int i=0;i<=n;++i) d[i][id]=1e9;
	d[x][id]=0;
	queue<int>q;
	q.push(x);
	vis[x]=true;
	while(!q.empty())
	{
		int top=q.front();
		q.pop();
		vis[top]=false;
		for(auto v:g[top])
		{
			if(d[top][id]+1<d[v][id])
			{
				d[v][id]=d[top][id]+1;
				if(!vis[v]) q.push(v),vis[v]=true;
			}
		}
	}
}
int main()
{
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d %d %d %d %d",&n,&m,&a,&b,&c);
		for(int i=0;i<=n;++i) g[i].clear();
		for(int i=1;i<=m;++i) scanf("%lld",&num[i]);
		sort(num+1,num+1+m);
		sum[0]=0;
		for(int i=1;i<=m;++i) sum[i]=sum[i-1]+num[i];
		for(int i=1;i<=m;++i)
		{
			scanf("%d %d",&u,&v);
			g[u].push_back(v);
			g[v].push_back(u);
		}
		spfa(a,0);spfa(b,1);spfa(c,2);
		ll ans=2*sum[m];
		for(int i=1;i<=n;++i)
		{
			int t1=d[i][1],t2=d[i][0]+d[i][2];
			if(t1+t2>m) continue;
			ans=min(ans,sum[t1]+sum[t1+t2]);
		}
		printf("%lld\n",ans);
	}
}