結果

問題 No.92 逃走経路
コンテスト
ユーザー むらため
提出日時 2019-01-22 02:01:44
言語 Nim
(2.2.0)
結果
AC  
実行時間 57 ms / 5,000 ms
コード長 983 bytes
コンパイル時間 4,049 ms
コンパイル使用メモリ 73,168 KB
実行使用メモリ 14,848 KB
最終ジャッジ日時 2024-07-01 10:30:53
合計ジャッジ時間 5,297 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sequtils,algorithm,strutils,intsets,tables
template times*(n:int,body) = (for _ in 0..<n: body)

proc getchar_unlocked():char {. importc:"getchar_unlocked",header: "<stdio.h>" .}
proc scan(): int =
  while true:
    let k = getchar_unlocked()
    if k < '0': break
    result = 10 * result + k.ord - '0'.ord

let n = scan()
let m = scan()
let k = scan()
var P = newTable[int,seq[seq[bool]]]()
m.times:
  let a = scan()
  let b = scan()
  let c = scan()
  if c notin P : P[c] = newSeqWith(n+1,newSeq[bool](n+1))
  P[c][a][b] = true
  P[c][b][a] = true
let D = newSeqWith(k,scan())
var answers = initIntSet()
let p0 = P[D[0]]
for a in 1..n:
  for b in (a+1)..n:
    if not p0[a][b] : continue
    answers.incl a
    answers.incl b
for i in 1..<k:
  let pn = P[D[i]]
  var nexts = initIntSet()
  for a in answers:
    for b in 1..n:
      if not pn[a][b] : continue
      nexts.incl b
  answers = nexts
echo answers.card
echo toSeq(answers.items).sorted(cmp).mapIt($it).join(" ")
0