Submission #2823629

---

Source Code Expand

N, M = gets.split.map(&:to_i)
x, a, mod = gets.split.map(&:to_i)
xs = Array.new(N*M)
xs[0] = x
(1...(N*M)).each do |i|
    xs[i] = (xs[i-1]+a) % mod
end
order = (0...(N*M)).sort_by{|i| xs[i]}

ans = 0
N.times do |i|
    arr = order[i*M, M].sort_by{|num| num % M}
    M.times do |j|
        ans += (arr[j]/M-i).abs + (arr[j]%M-j).abs
    end
end
puts ans

Submission Info

Judge Result