公交车站 --- lintcode825

点击打开原题

已知：

有各种各样的公交车路线，给定起点站点和终点站点，求最少换乘的次数，如果无法到达就返回-1

思路：

首先找到所有包含初始站点的公交车，再找到所有包含结束站点的公交车，将两者送入函数处理，获取最少转乘次数。如果找到转乘次数为1的，则直接返回1，如果找不到，就计算转乘次数为2的情况（假设从公交车a能转乘到b，则返回2）。如果还不行，则计算转乘为3的情况（假设包含七点的公交a能到达bcd，查看bcd中是否有能到包含终点的公交e的），依次类推

public class Solution {
    /**
     * @param N: The number of buses
     * @param route: The route of buses
     * @param A: Start bus station
     * @param B: End bus station
     * @return: Return the minimum transfer number
     */
       public int getMinTransferNumber(int N, int[][] route, int A, int B) {
		List<Integer> start = new ArrayList<Integer>();
		List<Integer> end = new ArrayList<Integer>();
		for (int i = 0; i < route.length; i++) {
			if (contain(route[i], A)) {
				start.add(i);
			}
			if (contain(route[i], B)) {
				end.add(i);
			}
		}

		int ret = 101;
    	        for (int i : start) {
			for (int j : end) {
				int temp = calculate(i, j, route);
				if (temp == -1) {
					continue;
				} else {
					if (temp == 1) {
						return 1;
					}
					if (ret > temp) {
						ret = temp;
					}
				}
			}
		}

		if (ret == 101) {
			return -1;
		}

		return ret;
	}

	private int calculate(int i, int j, int[][] route) {
		if (i == j) {
			return 1;
		}

		List<Integer> list = new ArrayList<Integer>();
		list.add(i);
		int start = 0;
		int end = 1;
		for (int ret = 2; ret <= 100; ret++) {

			for (int itemp = 0; itemp < route.length; itemp++) {
				if (list.contains(itemp)) {
					continue;
				}
				for (int jtemp = start; jtemp < end; jtemp++) {
					if (connect(route[itemp], route[list.get(jtemp)])) {
						list.add(itemp);
						break;
					}
				}
			}

			if (list.contains(j)) {
				return ret;
			}

			start = end;
			end = list.size();
			if (start == end) {
				return -1;
			}
		}

		return -1;
	}

	private boolean connect(int[] i, int[] j) {
		for (int num1 : i) {
			for (int num2 : j) {
				if (num1 == num2) {
					return true;
				}
			}
		}
		return false;
	}

	private boolean contain(int[] route, int site) {
		for (int i : route) {
			if (i == site) {
				return true;
			}
		}
		return false;
	}
}