Java——查找里程

课程：面向对象程序设计——Java语言（翁恺）

---

题目内容：

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

 

 

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

 

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

注意：你并不需要去录入上图的数据，数据是在程序输入中给的。

 

输入格式:

首先，你会读到若干个城市的名字。每个名字都只是一个英文单词，中间不含空格或其他符号。当读到名字为“###”（三个#号）时，表示城市名字输入结束，###并不是一个城市的名字。如果记读到的城市名字的数量为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.HashMap;
import java.util.Scanner;

public class Main{
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		
		//用ArrayList记录输入的城市，以###结束
		ArrayList<String> cities = new ArrayList<String>();
		String city = in.next();
		while( !city.equals("###")) {
			cities.add(city);
			city = in.next();
		}
		
		//用Distances对象记录输入的城市和对应的里程数
		int distance = in.nextInt();
		int num = cities.size();
		Distances allDistances = new Distances();
		for (int i=0; i<num; i++) {
			String cityA = cities.get(i);
			allDistances.addCity(cityA);
			for(int j=0; j<num; j++) {
				allDistances.addADistance(cityA, cities.get(j), distance);
				if(in.hasNextInt()) {
					distance = in.nextInt();
				}else{
					break;
				}
			}
		}
		
		//读取两个城市，输出两者间的里程数
		String city1 = in.next();
		String city2 = in.next();
		System.out.println(allDistances.getDistance(city1, city2));
		
		in.close();
	}

}

//构建Distances类，使用嵌套HashMap存储城市和里程数
class Distances{
	private HashMap<String, HashMap<String,Integer>> allDistances = new HashMap<String, HashMap<String, Integer>>();
	
	public void addCity(String city) {
		allDistances.put(city, new HashMap<String, Integer>());
	}
	
	public void addADistance(String cityA, String cityB, int dis) {
		allDistances.get(cityA).put(cityB, dis);
	}
	
	public int getDistance(String cityA, String cityB) {
		int distance = allDistances.get(cityA).get(cityB);
		return distance;
	}
}