牛客练习赛93 - 点权 —— 多源最短路

题目来源

---

题意：

给定 n 节点的一棵树，每条边有正边权 wi。
初始叶子节点已染色，其余节点未染色。
每次操作可以用两个已染色的节点将一个 与其相连的未染色节点 染色，花费为两条相连边的边权之和。
问对于每个点 i，其染色的最小花费。如果无法染色，输出 -1。

分析：

两个节点可以将一个节点更新。

所以对于每个点要找最小花费的两个节点来将其更新，用两个数组 min1[i], min2[i] 来表示用于其更新的两个最小花费的节点（加上边权）。

首先先将所有的叶子节点放到优先队列中，每取出一个点将其所有相邻未染色点的两个最小花费数组更新。如果相邻点的两个数组中都有值，说明其找到两个点了，那么就将其放到优先队列中。

类似于 dijkstra 求最短路，对于一个点，花费较小（距离较近）的点出队的时候会将该节点更新，然后将更新的答案放到优先队列中。
而该点一旦出队，表明已经没有花费更小的点能够将其更新了，也表明其最小花费（最短距离）就确定了。

于是便可以求出每个点的最小花费。

Code：

#include<bits/stdc++.h>
using namespace std;

#define Ios ios::sync_with_stdio(false),cin.tie(0)
#define mem(a,b) memset(a,b,sizeof a)
#define int long long
#define PII pair<int,int>
#define pb push_back
#define fi first
#define se second
#define endl '\n'
map<int,int> mp;

/**/

const int N = 200010, mod = 1e9+7;
int T, n, m;
int a[N], f[N], dist[N];
int ru[N];
vector<PII> e[N];
int min1[N], min2[N];

void bfs()
{
	priority_queue<PII, vector<PII>, greater<PII> > que;
	
	mem(dist, 0x3f);
	for(int i=1;i<=n;i++) if(ru[i] < 2) que.push({0, i}), dist[i] = 0;
	for(int i=1;i<=n;i++) min1[i] = min2[i] = 1e9;
	
	while(que.size())
	{
		int x = que.top().se, dis = que.top().fi;
		que.pop();
		if(f[x]) continue;
		
		f[x] = 1;
		dist[x] = dis;
		
		for(auto it:e[x])
		{
			int tx = it.fi, w = it.se;
			if(w+dist[x] < min1[tx]) min2[tx] = min1[tx], min1[tx] = w+dist[x];
			else if(w+dist[x] < min2[tx]) min2[tx] = w+dist[x];
			
			if(min2[tx] != 1e9) que.push({min1[tx]+min2[tx], tx});
		}
	}
}

signed main(){
	Ios;
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int x,y,z;cin>>x>>y>>z;
		e[x].pb({y, z});
		e[y].pb({x, z});
		ru[x]++, ru[y]++;
	}
	
	bfs();
	
	for(int i=1;i<=n;i++)
	{
		if(ru[i] < 2) cout<< 0 <<" ";
		else if(dist[i] < 1e9) cout<<dist[i]<<" ";
		else cout<<-1<<" ";
	}
	
	return 0;
}

如果不止两个点，用 x 个节点来更新的话，可以将每个点都开一个优先队列，找到花费最小的 x 个节点将其更新。

#include<bits/stdc++.h>
using namespace std;

#define Ios ios::sync_with_stdio(false),cin.tie(0)
#define mem(a,b) memset(a,b,sizeof a)
#define int long long
#define PII pair<int,int>
#define pb push_back
#define fi first
#define se second
#define endl '\n'
map<int,int> mp;

/**/

const int N = 200010, mod = 1e9+7;
int T, n, m;
int a[N], f[N], dist[N];
int ru[N];
vector<PII> e[N];

priority_queue<int, vector<int>, greater<int> > q[N];

void bfs()
{
	priority_queue<PII, vector<PII>, greater<PII> > que;
	
	mem(dist, 0x3f);
	for(int i=1;i<=n;i++) if(ru[i] < 2) que.push({0, i}), dist[i] = 0;
	
	while(que.size())
	{
		int x = que.top().se, dis = que.top().fi;
		que.pop();
		if(f[x]) continue;
		
		f[x] = 1;
		dist[x] = dis;
		
		for(auto it:e[x])
		{
			int tx = it.fi, w = it.se;
			q[tx].push(w+dist[x]);
			
			if(q[tx].size() >= 2){
				int t = q[tx].top();q[tx].pop();
				que.push({t+q[tx].top(), tx});
				q[tx].push(t);
			}
		}
	}
}

signed main(){
	Ios;
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int x,y,z;cin>>x>>y>>z;
		e[x].pb({y, z});
		e[y].pb({x, z});
		ru[x]++, ru[y]++;
	}
	
	bfs();
	
	for(int i=1;i<=n;i++)
	{
		if(ru[i] < 2) cout<< 0 <<" ";
		else if(dist[i] < 1e9) cout<<dist[i]<<" ";
		else cout<<-1<<" ";
	}
	
	return 0;
}

---

挺好的一道题~