#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; } |