CF295B Greg and Graph

题目传送门:CF295B

这道题是练习 floydfloyd 的好题。

首先我们来思考一下,删除 nn 次点,每次删点求一次最短路,用 floydfloyd 做则是 O(n3×n)\text O(n^3 \times n) 也就是 O(n4)\text O(n^4) 的复杂度,会 TT 飞,于是我们考虑从后往前,每次删点当作加点即可。优化后复杂度 O(n3)O(n^3),可以过。

代码:

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

#define int long long
const int N = 505;
int n;
int dis[N][N], del[N], ans[N];
bool f[N];

void floyd()
{
	for (int p = n; p >= 1; p--) 
    {
		int k = del[p];
		f[k] = true;
		int sum = 0;
		for (int i = 1; i <= n; i++)
		{
			for (int j = 1; j <= n; j++)
			{
				if (i != j)
				{
					dis[i][j] = min(dis[i][j], dis[i][k] + dis[k][j]);
					if (f[i] && f[j])
					{
						sum += dis[i][j];
					}
				}
			}
		}
		ans[p] = sum;
	}
}

void print()
{
	for (int i = 1; i <= n; i++)
	{
		cout << ans[i] << " ";
	}
	cout << "\n";
}

void input()
{
	cin >> n;
	memset(dis, 0x3f, sizeof(dis));
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			int w;
			cin >> w;
			dis[i][j] = w;
		}
	}
	for (int i = 1; i <= n; i++)
	{
		cin >> del[i];
	}
}

signed main()
{
	input();
	floyd();
	print();
	return 0;
}
posted @   HappyBobb  阅读(6)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示