ruby - 用 Ruby 编写这个的更好方法？

Question

我是 Ruby 新手。经过大量的重构，我归结为这个。有没有更好的写法？

 51   def tri_num?(n)
 52     i = 1
 53     while i < n
 54       return i if i * (i + 1) / 2 == n
 55       i += 1
 56     end 
 57     raise InvalidTree
 58   end

Answer 1

直接解决怎么样？

def tri_num? n

  i = (0.5*(-1.0 + Math.sqrt(1.0 + 8.0*n))).to_i

  if i*(i+1)/2 == n
    return i
  else
    raise InvalidTree
  end

end

虽然我不知道 tri_num? 是不是个好名字。通常以 ? 结尾的函数应返回 true 或 false。