結果

問題 No.990 N×Mマス計算（Kの倍数）
コンテスト
ユーザー siman
提出日時 2020-02-14 22:57:52
言語 Ruby
(3.4.1)
結果
WA  
実行時間 -
コード長 556 bytes
コンパイル時間 212 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 34,176 KB
最終ジャッジ日時 2024-11-16 01:18:34
合計ジャッジ時間 4,921 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 7 WA * 9 RE * 3
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

class Integer
  def mod_inverse(mod)
    self.pow(mod - 2, mod)
  end
end

N, M, K = gets.split.map(&:to_i)
op, *B = gets.chomp.split
B.map!(&:to_i)
A = N.times.map { gets.to_i }
ans = 0
mod_counter = Hash.new(0)
mod_counter_plus = Hash.new(0)

B.each do |b|
  mod_counter[b % (K - 1)] += 1
  mod_counter_plus[b % K] += 1
end

A.each do |a|
  if op == '+'
    mk = a % K
    ans += mod_counter_plus[K - mk]
  else
    if a % K == 0
      ans += M
    else
      ans += mod_counter[1] + mod_counter[K - 1 - a.mod_inverse(K - 1)]
    end
  end
end

puts ans
0