競プロおぼえがき

おぼえがきのためのブログ

yukicoder | No.92 逃走経路

  • 問題

No.92 逃走経路 - yukicoder

  • 概要

 N個の町があり。 M個の道の情報が渡される。
 i番目の道の情報  a_{i}, b_{i}, c_{i}は、 町a_{i}と町b_{i}をコストc_{i}で移動できる事を示している。
ここでK個の移動コスト履歴 d_{1} ... d_{K}が与えられたときに、移動完了後の町の位置としてあり得るものを列挙する。

  • 考え方

 n[K + 1][N + 1]の配列を使う。
最初の状態ではすべての町 iにいる可能性があるので n[0][i] = trueにする。
 k番目の移動をするとき、 k - 1番目までの移動でその町にいる可能性がある(n[k - 1][i] == true)かつ k番目の移動コストで移動できる道 mがあるときにその町のフラグを更新する。
(n[k][a[m]] or n[k][b[m]] = true)
 K番目までの移動が完了したときに行ける可能性のある町n[K][i]を数えて列挙する。

import java.util.Arrays;
import java.util.Scanner;

class Main {
    static Scanner in = new Scanner(System.in);

    public static void main(String[] args) {
        int N = in.nextInt(), M = in.nextInt(), K = in.nextInt();
        int[] a = new int[M + 1], b = new int[M + 1], c = new int[M + 1];
        for (int i = 1; i < M + 1; i++) {
            a[i] = in.nextInt();
            b[i] = in.nextInt();
            c[i] = in.nextInt();
        }
        int[] d = new int[K + 1];
        for(int i = 1; i < K + 1; i++){
            d[i] = in.nextInt();
        }
        boolean[][] n = new boolean[K + 1][N + 1];
        Arrays.fill(n[0], true);
        for(int i = 1; i < K + 1; i++){
            Arrays.fill(n[i], false);
        }
        
        for(int x = 1; x < K + 1; x++){
            for(int y = 1; y < N + 1; y++){
                for(int z = 1; z < M + 1; z++){
                    if(n[x - 1][y]){
                        if(a[z] == y && c[z] == d[x]){
                            n[x][b[z]] = true;
                        }
                        if(b[z] == y && c[z] == d[x]){
                            n[x][a[z]] = true;
                        }
                    }
                }
            }
        }
        int sum = 0;
        for(int i = 1; i < N + 1; i++){
            if(n[K][i]){
                sum++;
            }
        }
        System.out.println(sum);
        boolean t = true;
        for(int i = 1; i < N + 1; i++){
            if(t && n[K][i]){
                System.out.print(i);
                t = false;
            }else if(n[K][i]){
                System.out.print(" " + i);
            }
        }
        System.out.println();
    }
}