結果

問題 No.762 PDCAパス
コンテスト
ユーザー TANIGUCHI Kousuke
提出日時 2019-12-18 11:39:12
言語 Ruby
(3.4.1)
結果
WA  
実行時間 -
コード長 534 bytes
コンパイル時間 37 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 26,752 KB
最終ジャッジ日時 2024-07-06 03:57:42
合計ジャッジ時間 8,810 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36 WA * 2
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

TYPES = "PDCA"

N,M = gets.split.map(&:to_i)
S = gets.chomp
V = Array.new(N + 1, -1)
G = Array.new(N + 1){[]}
dp = Array.new(4){ Array.new(N + 1, 0) }

(1 .. N).each do |v|
  t = TYPES.index(S[v - 1])
  V[v] = t
end

M.times do 
  a,b = gets.split.map(&:to_i)
  G[a] << b
  G[b] << a
end

(1 .. N).each do |v|
  dp[0][v] = 1 if V[v] == 0
end

(1 .. 3).each do |t|
  (1 .. N).each do |v|
    next unless V[v] == t
    
    dp[t][v] = G[v].inject(0) do |s, u|
      V[u] == t - 1 ? s + dp[t - 1][u] : s
    end
  end
end
puts dp[3].sum
0