結果

問題 No.990 N×Mマス計算(Kの倍数)
ユーザー siman
提出日時 2020-02-14 21:56:29
言語 Ruby
(3.4.1)
結果
WA  
実行時間 -
コード長 471 bytes
コンパイル時間 260 ms
コンパイル使用メモリ 7,296 KB
実行使用メモリ 29,440 KB
最終ジャッジ日時 2024-11-16 00:38:42
合計ジャッジ時間 4,525 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1 WA * 1
other AC * 9 WA * 10
権限があれば一括ダウンロードができます
コンパイルメッセージ
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)

B.each do |b|
  mod_counter[b % K] += 1
end

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

puts ans
0