結果

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

ソースコード

diff #

TYPES = "PDCA"
MOD = 10**9 + 7

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].inject(0){|s,v| (s + v) % MOD }
0