More than 5 years have passed since last update.

Ruby の sum は地味にすごい話

Last updated at 2020-11-02Posted at 2020-11-02

はじめに

速い速いと評判のsumが本当に速かった話

今回のテーマ、等差数列の和

題意としては、普通に加算するとTLEするので、等差数列の和の公式

S=n(a+b)/2

を使いましょうというもの。

ruby.rb

n = gets.to_i
sum = 0
n.times do
  a, b = gets.split.map(&:to_i)
  sum += (a..b).inject(:+)
end
puts sum

見事にTLE

ruby.rb

n = gets.to_i
sum = 0
n.times do
  a, b = gets.split.map(&:to_i)
  sum += (a..b).sum
end
puts sum

普通に通ります。
rangeのsumは内部的に、等差数列の和の公式を使っていると思われます~~（適当）~~
どうやら本当だった様子。
コメント欄を参照願います。