結果

問題 No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
コンテスト
ユーザー universato
提出日時 2020-05-30 07:11:36
言語 Ruby
(4.0.2)
コンパイル:
ruby -w -c _filename_
実行:
ruby _filename_
結果
AC  
実行時間 1,804 ms / 2,000 ms
コード長 361 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 76 ms
コンパイル使用メモリ 9,216 KB
実行使用メモリ 15,744 KB
最終ジャッジ日時 2026-05-07 08:15:01
合計ジャッジ時間 13,416 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge3_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.rb:3: warning: assigned but unused variable - q
Syntax OK

ソースコード

diff #
raw source code 

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+2){ 0 }
dp[0] = 1

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

b.map!{|t| dp[t] < mod ? dp[t] : dp[t] % mod }

puts b.join("\n")
0