結果

問題 No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
ユーザー universato
提出日時 2020-05-29 22:41:35
言語 Ruby
(3.4.1)
結果
MLE  
実行時間 -
コード長 426 bytes
コンパイル時間 381 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 290,724 KB
最終ジャッジ日時 2024-11-06 06:33:04
合計ジャッジ時間 6,909 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 7 MLE * 1 -- * 16
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.rb:3: warning: assigned but unused variable - q
Syntax OK

ソースコード

diff # 

mod = 998244353

n, q = gets.to_s.split.map{|t| t.to_i }
a    = gets.to_s.split.map{|t| t.to_i }
b    = gets.to_s.split.map{|t| t.to_i }

dp = Array.new(n+1){ [0] * (n+2) }
dp[0][0] = 1

a.each_with_index do |t, i|
  
  i.downto(0) do |j|
    dp_cur = dp[i][j]
    dp[i+1][j] = ( dp[i+1][j] + dp_cur * (t - 1) ) % mod
    dp[i+1][j+1] = ( dp[i+1][j+1] + dp_cur ) % mod
  end
end

b.map!{|t| dp[n][t] % mod }

puts b.join("\n")
0