結果

問題 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 + " ");
            }
        }
    }
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0