Codeforces Round 977 (Div. 2)E1 Digital Village (Easy Version)(Floyd,贪心)

题目链接

Codeforces Round 977 (Div. 2)E1 Digital Village (Easy Version)

思路

首先,我们注意到 n n n的最大值只有 400 400 400。

因此,我们可以先用 F l o y d Floyd Floyd算法预处理出任意两座城市之间的最大延迟时间。

之后,我们通过在线操作,每次贪心地选出最优的一个城市,并不断更新答案。

即,我们先选出 k = 1 k=1 k=1时的最优解,之后从剩下的点里面挑出一个能够使 k = 2 k=2 k=2时最优的点,以此类推。

代码

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 4e2 + 5;
const int mod = 998244353;
const int inf = 0x3f3f3f3f3f3f3f3f;
int n, m, p;
void solve()
{
	cin >> n >> m >> p;
	vector<int>city(p);
	for (int &val : city)
	{
		cin >> val;
	}
	vector<vector<int>>dist(n + 1, vector<int>(n + 1, inf));
	for (int i = 1, u, v, w; i <= m; i++)
	{
		cin >> u >> v >> w;
		dist[u][v] = min(dist[u][v], w);
		dist[v][u] = min(dist[v][u], w);
	}

	for (int i = 1; i <= n; i++)
		dist[i][i] = 0;

	for (int k = 1; k <= n; k++)
	{
		for (int i = 1; i <= n; i++)
		{
			for (int j = 1; j <= n; j++)
			{
				dist[i][j] = min(dist[i][j], max(dist[i][k], dist[k][j]));
			}
		}
	}


	vector<int>ans(n + 1);

	int sum = inf, idx = -1;
	for (auto u : city)
	{
		int res = 0;
		for (auto v : city)
		{
			res += dist[u][v];
		}
		if (sum >= res)
		{
			idx = u;
			sum = res;
			for (auto v : city)
			{
				ans[v] = dist[u][v];
			}
		}
	}

	set<int>st;
	st.insert(idx);

	cout << sum << " ";

	for (int i = 2; i <= n; i++)
	{
		if (sum == 0)
		{
			cout << sum << " ";
			continue;
		}
		int maxx = 0;
		idx = -1;
		for (auto u : city)
		{
			if (st.count(u)) continue;
			int res = 0;
			for (auto v : city)
			{
				if (st.count(v)) continue;
				if (ans[v] > dist[u][v])
				{
					res += ans[v] - dist[u][v];
				}
			}
			if (maxx <= res)
			{
				maxx = res;
				idx = u;
			}
		};
		st.insert(idx);
		sum -= maxx;
		for (auto v : city)
		{
			if (st.count(v)) continue;
			ans[v] = min(ans[v], dist[idx][v]);
		}
		cout << sum << " ";
	}
	cout << endl;
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int test = 1;
	cin >> test;
	for (int i = 1; i <= test; i++)
	{
		solve();
	}
	return 0;
}
相关推荐
ragnwang32 分钟前
C++ Eigen常见的高级用法 [学习笔记]
c++·笔记·学习
yuanManGan1 小时前
数据结构漫游记:静态链表的实现(CPP)
数据结构·链表
火星机器人life2 小时前
基于ceres优化的3d激光雷达开源算法
算法·3d
虽千万人 吾往矣2 小时前
golang LeetCode 热题 100(动态规划)-更新中
算法·leetcode·动态规划
arnold663 小时前
华为OD E卷(100分)34-转盘寿司
算法·华为od
ZZTC3 小时前
Floyd算法及其扩展应用
算法
lqqjuly4 小时前
特殊的“Undefined Reference xxx“编译错误
c语言·c++
lshzdq4 小时前
【机器人】机械臂轨迹和转矩控制对比
人工智能·算法·机器人
冰红茶兑滴水4 小时前
云备份项目--工具类编写
linux·c++
刘好念4 小时前
[OpenGL]使用 Compute Shader 实现矩阵点乘
c++·计算机图形学·opengl·glsl