結果

問題 No.1 道のショートカット
ユーザー maruyuki95
提出日時 2020-05-18 17:42:02
言語 Java
(openjdk 23)
結果
AC  
実行時間 246 ms / 5,000 ms
コード長 2,439 bytes
コンパイル時間 2,378 ms
コンパイル使用メモリ 78,128 KB
実行使用メモリ 46,140 KB
最終ジャッジ日時 2024-07-20 16:48:10
合計ジャッジ時間 11,426 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff #

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

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);

		int cityNum = sc.nextInt();
		int money = sc.nextInt();
		int roadNum = sc.nextInt();

		List<Integer> cityS = new ArrayList<Integer>();
		for (int i = 0; i < roadNum; i++) {
			cityS.add(sc.nextInt());
		}

		List<Integer> cityT = new ArrayList<Integer>();
		for (int i = 0; i < roadNum; i++) {
			cityT.add(sc.nextInt());
		}

		List<Integer> costs = new ArrayList<Integer>();
		for (int i = 0; i < roadNum; i++) {
			costs.add(sc.nextInt());
		}

		List<Integer> times = new ArrayList<Integer>();
		for (int i = 0; i < roadNum; i++) {
			times.add(sc.nextInt());
		}

		int ret = new Main().execute(cityNum, money, roadNum, cityS, cityT, costs, times);
		System.out.println(ret);
	}

	private int execute(int cityNum, int money, int roadNum, List<Integer> cityS, List<Integer> cityT, List<Integer> costs, List<Integer> times) {
		searchRoute(1, cityNum, cityS, cityT, money, costs, 0, times);
		if(minTime == null) {
			return -1;
		}
		return minTime.intValue();
	}

	private Integer minTime = null;
	private void searchRoute(int location, int goal, List<Integer> cityS, List<Integer> cityT, int money, List<Integer> costs, int totalTime, List<Integer> times) {
		List<Integer> roadNums = filterRoadNumsWithLocation(cityS, location);		// 道の入り口が現在地と同じ道のみに絞る

		for (Integer roadNum : roadNums) {
			Integer cost = costs.get(roadNum);
			if (money < cost) {	// 資金が足りないのでスキップ
				continue;
			}

			int afterTime = totalTime + times.get(roadNum);
			if (minTime != null && minTime.intValue() <= afterTime) { // 既に最短時間よりも時間がかかっているためスキップ
				continue;
			}

			Integer afterLocation = cityT.get(roadNum);
			if (goal == afterLocation) {
				minTime = afterTime;
				continue;
			}else {	// 道の出口が目的地ではないため、再帰して探索
				int afterMoney = money - cost;
				searchRoute(afterLocation, goal, cityS, cityT, afterMoney, costs, afterTime, times);
			}
		}
	}

	private List<Integer> filterRoadNumsWithLocation(List<Integer> cityS, int location) {
		List<Integer> ret = new ArrayList<Integer>();
		for (int i = 0; i < cityS.size(); i++) {
			if (cityS.get(i) == location) {
				ret.add(i);
			}
		}
		return ret;
	}

}
0