#include<iostream>
#include<queue>
#define INF 500
bool edge[400][400]; // [from][to]
int distance[400][400]; // [from][to]
int size;
struct CategoryEdge {
int from;
int to;
};
CategoryEdge categories[400][160000]; // [distance][index]
int categorySizes[400]; // [category]
void BFS(int start) {
std::queue<int> q;
for (int i = 0; i < size; i++) {
distance[start][i] = INF;
}
distance[start][start] = 0;
q.push(start);
while (!q.empty()) {
int current = q.front();
q.pop();
for (int i = 0; i < size; i++) {
if (edge[current][i] && distance[start][i] == INF) {
distance[start][i] = distance[start][current] + 1;
q.push(i);
}
}
}
}
int getDiameterWithTeleport(int teleport1, int teleport2, int upperBound) {
int diameter = 0;
for (int i = size - 1; i > diameter; i--) {
for (int j = 0; j < categorySizes[i]; j++) {
int from = categories[i][j].from;
int to = categories[i][j].to;
diameter = std::max(diameter, std::min(distance[from][to], std::min(distance[from][teleport1] + distance[to][teleport2], distance[from][teleport2] + distance[to][teleport1])));
if (diameter >= upperBound) {
return upperBound;
}
}
}
return diameter;
}
int main() {
std::ios_base::sync_with_stdio(false);
std::cin.tie(NULL);
int t;
std::cin >> t;
for (int i = 0; i < t; i++) {
std::cin >> size;
for (int j = 0; j < size; j++) {
for (int k = 0; k < size; k++) {
char c;
std::cin >> c;
edge[j][k] = c == '1';
}
}
for (int j = 0; j < size; j++) {
BFS(j);
}
for (int j = 0; j < size; j++) {
categorySizes[j] = 0;
}
for (int j = 0; j < size; j++) {
for (int k = 0; k < j; k++) {
categories[distance[j][k]][categorySizes[distance[j][k]]].from = j;
categories[distance[j][k]][categorySizes[distance[j][k]]].to = k;
categorySizes[distance[j][k]]++;
}
}
int minDiameter = INF;
for (int j = 0; j < size; j++) {
for (int k = 0; k < j; k++) {
minDiameter = getDiameterWithTeleport(j, k, minDiameter);
}
}
std::cout << minDiameter << "\n";
}
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 | #include<iostream> #include<queue> #define INF 500 bool edge[400][400]; // [from][to] int distance[400][400]; // [from][to] int size; struct CategoryEdge { int from; int to; }; CategoryEdge categories[400][160000]; // [distance][index] int categorySizes[400]; // [category] void BFS(int start) { std::queue<int> q; for (int i = 0; i < size; i++) { distance[start][i] = INF; } distance[start][start] = 0; q.push(start); while (!q.empty()) { int current = q.front(); q.pop(); for (int i = 0; i < size; i++) { if (edge[current][i] && distance[start][i] == INF) { distance[start][i] = distance[start][current] + 1; q.push(i); } } } } int getDiameterWithTeleport(int teleport1, int teleport2, int upperBound) { int diameter = 0; for (int i = size - 1; i > diameter; i--) { for (int j = 0; j < categorySizes[i]; j++) { int from = categories[i][j].from; int to = categories[i][j].to; diameter = std::max(diameter, std::min(distance[from][to], std::min(distance[from][teleport1] + distance[to][teleport2], distance[from][teleport2] + distance[to][teleport1]))); if (diameter >= upperBound) { return upperBound; } } } return diameter; } int main() { std::ios_base::sync_with_stdio(false); std::cin.tie(NULL); int t; std::cin >> t; for (int i = 0; i < t; i++) { std::cin >> size; for (int j = 0; j < size; j++) { for (int k = 0; k < size; k++) { char c; std::cin >> c; edge[j][k] = c == '1'; } } for (int j = 0; j < size; j++) { BFS(j); } for (int j = 0; j < size; j++) { categorySizes[j] = 0; } for (int j = 0; j < size; j++) { for (int k = 0; k < j; k++) { categories[distance[j][k]][categorySizes[distance[j][k]]].from = j; categories[distance[j][k]][categorySizes[distance[j][k]]].to = k; categorySizes[distance[j][k]]++; } } int minDiameter = INF; for (int j = 0; j < size; j++) { for (int k = 0; k < j; k++) { minDiameter = getDiameterWithTeleport(j, k, minDiameter); } } std::cout << minDiameter << "\n"; } return 0; } |
English