結果

問題 No.92 逃走経路
ユーザー YamaKasa
提出日時 2018-07-07 13:36:34
言語 Java
(openjdk 23)
結果
AC  
実行時間 251 ms / 5,000 ms
コード長 1,623 bytes
コンパイル時間 2,123 ms
コンパイル使用メモリ 85,520 KB
実行使用メモリ 47,448 KB
最終ジャッジ日時 2024-07-04 01:38:55
合計ジャッジ時間 7,304 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.Scanner;
import java.util.Set;
import java.util.TreeSet;

public class Main  {
	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		int N = scan.nextInt();
		int M = scan.nextInt();
		int K = scan.nextInt();
		int []a = new int[M];
		int []b = new int[M];
		int []c = new int[M];
		int []d = new int[K];
		for(int i = 0; i < M; i++) {
			a[i] = scan.nextInt();
			b[i] = scan.nextInt();
			c[i] = scan.nextInt();
		}
		for(int i = 0; i < K; i++) {
			d[i] = scan.nextInt();
		}
		scan.close();

		int [][]dp = new int[N + 1][K + 1];
		for(int i = 0; i <= N; i++) {
			for(int j = 0; j <= K; j++) {
				dp[i][j] = 0;
			}
		}

		// 1回目に犯人いる可能性のある街をdp[街][1回目] = 1とする。
		for(int i = 0; i < M; i++) {
			if(c[i] == d[0]) {
				dp[a[i]][1] = 1;
				dp[b[i]][1] = 1;
			}
		}

		// 2回目以降を調べる
		// c[i] = d[i] のとき、i - 1回目に犯人がいた可能性があるとき、i回目に
		// 犯人がいた可能性が存在する。
		for(int i = 1; i < K; i++) {
			for(int j = 0; j < M; j++) {
				if(c[j] == d[i]) {
					if(dp[a[j]][i] == 1) {
						dp[b[j]][i + 1] = 1;
					}
					if(dp[b[j]][i] == 1) {
						dp[a[j]][i + 1] = 1;
					}
				}
			}
		}

		// dp[][K] = 1となっている街を調べる
		Set<Integer> set = new TreeSet<Integer>();
		for(int i = 1; i <= N; i++) {
			if(dp[i][K] == 1) {
				set.add(i);
			}
		}
		System.out.println(set.size());
		int cnt = 0;
		for(int i : set) {
			if(cnt == set.size() - 1) {
				System.out.println(i);
			}else {
				System.out.print(i + " ");
			}
		}
	}
}
0