题目:http://pta.patest.cn/pta/test/15/exam/4/question/716
PTA - 数据结构与算法题目集(中文) - 5-8
哈利·波特要考试了,他需要你的帮助。这门课学的是用魔咒将一种动物变成另一种动物的本事。例如将猫变成老鼠的魔咒是haha,将老鼠变成鱼的魔咒是hehe等等。反方向变化的魔咒就是简单地将原来的魔咒倒过来念,例如ahah可以将老鼠变成猫。另外,如果想把猫变成鱼,可以通过念一个直接魔咒lalala,也可以将猫变老鼠、老鼠变鱼的魔咒连起来念:hahahehe。
现在哈利·波特的手里有一本教材,里面列出了所有的变形魔咒和能变的动物。老师允许他自己带一只动物去考场,要考察他把这只动物变成任意一只指定动物的本事。于是他来问你:带什么动物去可以让最难变的那种动物(即该动物变为哈利·波特自己带去的动物所需要的魔咒最长)需要的魔咒最短?例如:如果只有猫、鼠、鱼,则显然哈利·波特应该带鼠去,因为鼠变成另外两种动物都只需要念4个字符;而如果带猫去,则至少需要念6个字符才能把猫变成鱼;同理,带鱼去也不是最好的选择。
输入格式:
输入说明:输入第1行给出两个正整数N (≤100)和M,其中N是考试涉及的动物总数,M是用于直接变形的魔咒条数。为简单起见,我们将动物按1~N编号。随后M行,每行给出了3个正整数,分别是两种动物的编号、以及它们之间变形需要的魔咒的长度(≤100),数字之间用空格分隔。
输出格式:
输出哈利·波特应该带去考场的动物的编号、以及最长的变形魔咒的长度,中间以空格分隔。如果只带1只动物是不可能完成所有变形要求的,则输出0。如果有若干只动物都可以备选,则输出编号最小的那只。
输入样例:
6 11
3 4 70
1 2 1
5 4 50
2 6 50
5 6 60
1 3 70
4 6 60
3 6 80
5 1 100
2 4 60
5 2 80
输出样例:
4 70
package harry.potter;
import java.util.Scanner;
public class Demo1 {
static Scanner sc = null;
public static void main(String[] args) {
sc = new Scanner(System.in);
int N = sc.nextInt();
int M = sc.nextInt();
int[][] G = new int[N][N];
// 得到邻接矩阵
getAdjacentMatrix(G, M);
// 调用floyd算法
floyd(G);
// //打印出G矩阵
// for(int i=0;i<N;i++){
// for(int j=0;j<N;j++){
// System.out.print(G[i][j]+" ");
//
// }
// System.out.println();
// }
// 接下来找到G矩阵中每行的最大值,存入矩阵dist[N]
int[] dist = new int[N];
findLineMaxNum(G, dist);
int minAnimalNum = findMinNum(dist) + 1;
System.out.println("最小动物是" + minAnimalNum + ";最长变形魔咒长度是:" + dist[minAnimalNum - 1]);
sc.close();
}
private static int findMinNum(int[] dist) {
int N = dist.length;
// 把minNum设置成一个很大的数
int minNum = 10000;
int minAnimalNum = 0;
for (int i = 0; i < N; i++) {
if (minNum > dist[i]) {
minNum = dist[i];
minAnimalNum = i;
}
}
System.out.println(minAnimalNum);
return minAnimalNum;
}
public static void findLineMaxNum(int[][] G, int[] dist) {
int maxNum = 0;
int N = G.length;
for (int i = 0; i < N; i++) {
//每次循环maxNum都必须重新置为0
maxNum = 0;
for (int j = 0; j < N; j++) {
if (G[i][j] > maxNum) {
maxNum = G[i][j];
}
}
dist[i] = maxNum;
}
}
// floyd算法
private static void floyd(int[][] G) {
int N = G.length;
for (int k = 0; k < N; k++) {
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
if (G[i][k] + G[k][j] < G[i][j]) {
//注意:因为是无向的,所以是对称矩阵!
G[i][j] = G[i][k] + G[k][j];
G[j][i] = G[i][k] + G[k][j];
}
}
}
}
}
public static void getAdjacentMatrix(int[][] G, int M) {
int N = G.length;
// 把矩阵初始值设为一个很大的值,这里选用100
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
if (i != j) {
G[i][j] = 10000;
}
}
}
// 生成邻接矩阵
int m = 0;
int n = 0;
int num = 0;
for (int i = 0; i < M; i++) {
m = sc.nextInt() - 1;
n = sc.nextInt() - 1;
num = sc.nextInt();
G[m][n] = num;
G[n][m] = num;
}
}
}