题目:
一个公司在全国有 n
个分部,它们之间有的有道路连接。一开始,所有分部通过这些道路两两之间互相可以到达。公司意识到在分部之间旅行花费了太多时间,所以它们决定关闭一些分部(也可能不关闭任何分部),同时保证剩下的分部之间两两互相可以到达且最远距离不超过 maxDistance
。
两个分部之间的 距离 是通过道路长度之和的 最小值 。给你整数 n
,maxDistance
和下标从 0 开始的二维整数数组 roads
,其中 roads[i] = [ui, vi, wi]
表示一条从 ui
到 vi
长度为 wi
的 无向 道路。
请你返回关闭分部的可行方案数目,满足每个方案里剩余分部之间的最远距离不超过 maxDistance
。注意,关闭一个分部后,与之相连的所有道路不可通行。注意,两个分部之间可能会有多条道路。
示例 1:
输入:n = 3, maxDistance = 5, roads = [[0,1,2],[1,2,10],[0,2,10]]
输出:5
解释:可行的关闭分部方案有:
- 关闭分部集合 [2] ,剩余分部为 [0,1] ,它们之间的距离为 2 。
- 关闭分部集合 [0,1] ,剩余分部为 [2] 。
- 关闭分部集合 [1,2] ,剩余分部为 [0] 。
- 关闭分部集合 [0,2] ,剩余分部为 [1] 。
- 关闭分部集合 [0,1,2] ,关闭后没有剩余分部。
总共有 5 种可行的关闭方案。
示例 2:
输入:n = 3, maxDistance = 5, roads = [[0,1,20],[0,1,10],[1,2,2],[0,2,2]]
输出:7
解释:可行的关闭分部方案有:
- 关闭分部集合 [] ,剩余分部为 [0,1,2] ,它们之间的最远距离为 4 。
- 关闭分部集合 [0] ,剩余分部为 [1,2] ,它们之间的距离为 2 。
- 关闭分部集合 [1] ,剩余分部为 [0,2] ,它们之间的距离为 2 。
- 关闭分部集合 [0,1] ,剩余分部为 [2] 。
- 关闭分部集合 [1,2] ,剩余分部为 [0] 。
- 关闭分部集合 [0,2] ,剩余分部为 [1] 。
- 关闭分部集合 [0,1,2] ,关闭后没有剩余分部。
总共有 7 种可行的关闭方案。
提示:
1 <= n <= 10
1 <= maxDistance <= 105
0 <= roads.length <= 1000
roads[i].length == 3
0 <= ui, vi <= n - 1
ui != vi
1 <= wi <= 1000
- 一开始所有分部之间通过道路互相可以到达。
思路:
在不考虑时间复杂度的情况下,直接暴力算法思路还是很清晰的
由于要求取所有可能的方案数,所以我们枚举所有可能的分部开放方案,然后验证这个方案是否合理即可。
一个方案合不合理,要看任意两点之间的最短距离是否 < maxDistance,任意两点之间的最短距离,直接通过弗洛伊德算法求取就可以了。
代码:
class Solution {
public:
int numberOfSets(int n, int maxDistance, vector<vector<int>>& roads) {
int res = 0;
// 邻接矩阵表示图, 元素为(i, j)之间的距离
vector<vector<int>> d(n, vector<int>(n, 1000000));
// 表示n个分部的开放情况,为1就是开放
vector<int> opened(n, 0);
// 枚举二进制 0 - (1 << n) 表示所有可能的分部开放情况
for (int mask = 0; mask < (1 << n); mask++) {
// 用mask 给 opened赋值
for (int i = 0; i < n; i++) {
opened[i] = mask & (1 << i);
}
fill(d.begin(), d.end(), vector<int>(n, 1000000));
// 初始化邻接矩阵距离
for (const auto& road : roads) {
int i = road[0], j = road[1], r = road[2];
if (opened[i] > 0 && opened[j] > 0) {
d[i][j] = d[j][i] = min(d[i][j], r);
}
}
// 弗洛伊德算法求取任意两个点间的最短距离
for (int k = 0; k < n; k++) {
if (opened[k] > 0) {
for (int i = 0; i < n; i++) {
if (opened[i] > 0) {
for (int j = i + 1; j < n; j++) {
if (opened[j] > 0) {
d[i][j] = d[j][i] = min(d[i][j], d[i][k] + d[k][j]);
}
}
}
}
}
}
// 验证此方案是否有效
int good = 1;
for (int i = 0; i < n; i++) {
if (opened[i] > 0) {
for (int j = i + 1; j < n; j++) {
if (opened[j] > 0 && d[i][j] > maxDistance) {
good = 0;
break;
}
}
if (!good) {
break;
}
}
}
res += good;
}
return res;
}
};