java进阶-第三周编程题

1查找里程

（10分）

题目内容：

下图为国内主要城市之间的公路里程：

image

你的程序要读入这样的一张表，然后，根据输入的两个城市的名称，给出这两个城市之间的里程。

注意：任何两个城市之间的里程都已经给出，不需要计算经第三地中转。

输入格式:

首先，你会读到若干个城市的名字。每个名字都只是一个英文单词，中间不含空格或其他符号。当读到名字为“###”（三个#号）时，表示城市名字输入结束，###并不是一个城市的名字。如果记读到的城市名字的数量为n。

然后，你会读到nxn的一个整数矩阵。第一行的每一个数字，表示上述城市名单中第一个城市依次到另一个城市之间的里程。表中同一个城市之间的里程为0。

最后，你会读到两个城市的名字。

输出格式：

输出这两个城市之间的距离。

输入样例：

Hagzou Hugzou Jigxng ###

0 1108 708

1108 0 994

708 994 0

Hagzou Jigxng

输出样例：

708

时间限制：500ms

内存限制：32000kb

我的答案

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        ArrayList <String> citys = new ArrayList <String>();
        Scanner in = new Scanner(System.in);
        //读入城市  
        while (true) {
            String city = in.next();
            if (city.equals("###")) {
                break;
            }
            else
            {
                citys.add(city);
            }
        }
        // 读入里程表
        int cnt = citys.size();
        int [][] miles = new int[cnt][cnt];
        for (int i=0;i<cnt;i++) {
            for (int j=0;j<cnt;j++) {
                miles[i][j] = in.nextInt();
            }
        }
        // 返回里程
        String c1 = in.next();
        String c2 = in.next();
        int a = citys.indexOf(c1);
        int b = citys.indexOf(c2);
        System.out.println(miles[a][b]);
    }

}