結果

問題 No.92 逃走経路
ユーザー 👑 hos.lyric
提出日時 2019-01-10 02:11:05
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 7 ms / 5,000 ms
コード長 2,149 bytes
コンパイル時間 7,800 ms
コンパイル使用メモリ 315,184 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-13 02:28:53
合計ジャッジ時間 8,744 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.conv, std.stdio, std.string;
import std.algorithm, std.array, std.bigint, std.container, std.math, std.range, std.regex, std.typecons;
import core.bitop;

class EOFException : Throwable { this() { super("EOF"); } }
string[] tokens;
string readToken() { for (; tokens.empty; ) { if (stdin.eof) throw new EOFException; tokens = readln.split; } auto token = tokens[0]; tokens.popFront; return token; }
int readInt() { return readToken().to!int; }
long readLong() { return readToken().to!long; }
real readReal() { return readToken().to!real; }

void chmin(T)(ref T t, in T f) { if (t > f) t = f; }
void chmax(T)(ref T t, in T f) { if (t < f) t = f; }

int binarySearch(T)(in T[] as, in bool delegate(T) test) { int low = -1, upp = cast(int)(as.length); for (; low + 1 < upp; ) { int mid = (low + upp) >> 1; (test(as[mid]) ? low : upp) = mid; } return upp; }
int lowerBound(T)(in T[] as, in T val) { return as.binarySearch((T a) => (a < val)); }
int upperBound(T)(in T[] as, in T val) { return as.binarySearch((T a) => (a <= val)); }


int N, M, K;
int[] A, B, C;
int[] D;

void main() {
  try {
    for (; ; ) {
      N = readInt();
      M = readInt();
      K = readInt();
      A = new int[M];
      B = new int[M];
      C = new int[M];
      foreach (i; 0 .. M) {
        A[i] = readInt() - 1;
        B[i] = readInt() - 1;
        C[i] = readInt();
      }
      D = new int[K];
      foreach (k; 0 .. K) {
        D[k] = readInt();
      }
      
      auto graph = new int[][int][N];
      foreach (i; 0 .. M) {
        graph[A[i]][C[i]] ~= B[i];
        graph[B[i]][C[i]] ~= A[i];
      }
      
      auto dp = new bool[][](K + 1, N);
      dp[0][] = true;
      foreach (k; 0 .. K) {
        foreach (u; 0 .. N) {
          if (dp[k][u]) {
            foreach (v; graph[u].get(D[k], [])) {
              dp[k + 1][v] = true;
            }
          }
        }
      }
      
      int[] ans;
      foreach (u; 0 .. N) {
        if (dp[K][u]) {
          ans ~= u;
        }
      }
      writeln(ans.length);
      writeln(ans.map!(u => u + 1).to!string.replaceAll(regex(`[\[\],]`), ""));
    }
  } catch (EOFException e) {
  }
}
0