Ruby - ニュートン補間！

前回は、C++ による「ニュートン補間」のアルゴリズムを紹介しました。

• C++ - ニュートン補間！

今日は、同じアルゴリズムを Ruby で実現してみました。
アルゴリズムについては、上記リンクの記事を参照してください。

実際、ほとんど同じです。

以下、Ruby によるサンプルスクリプトです。

0. 前提条件

• Linux Mint 14 Nadia (64bit) での作業を想定。
• Ruby 2.0.0-p0 を使用。
• ニュートン補間そのものについての詳細は割愛。

1. Ruby スクリプト作成

File: interpolate_newton.rb

#! /usr/local/bin/ruby
#*********************************************
# ニュートン補間
#*********************************************
#
class InterpolateNewton
  # あらかじめ与える点
  X = [0.0, 2.0, 3.0, 5.0, 8.0]
  Y = [0.8, 3.2, 2.8, 4.5, 1.9]

  def initialize
    @n = X.size  # 点の数
  end

  # 計算・結果出力
  def calc
    puts "      x      y"
    X[0].step(X[@n - 1], 0.5) do |t|
      printf("%7.2f%7.2f\n", t, interpolate_newton(t))
    end
  end

  private

  # ニュートン補間
  def interpolate_newton(t)
    c = Array.new  # 係数配列
    w = Array.new  # 作業用配列
    # 差分商
    0.upto(@n - 1) do |i|
      w[i] = Y[i]
      (i - 1).downto(0) {|j| w[j] = (w[j + 1] - w[j]) / (X[i] - X[j])}
      c[i] = w[0]
    end
    # 総和
    s = c[@n - 1]
    (@n - 2).downto(0) {|i| s = s * (t - X[i]) + c[i]}
    return s
  end
end

if __FILE__ == $0
  begin
    # 計算クラスインスタンス化
    obj = InterpolateNewton.new
    # ニュートン補間計算
    obj.calc
  rescue => e
    $stderr.puts "[例外発生] #{e}"
  end
end

• Gist - Ruby script to interpolate with Newton method.

2. 実行

$ ./interpolate_newton.rb
      x      y
   0.00   0.80
   0.50   2.49
   1.00   3.23
   1.50   3.37
   2.00   3.20
   2.50   2.95
   3.00   2.80
   3.50   2.85
   4.00   3.17
   4.50   3.74
   5.00   4.50
   5.50   5.32
   6.00   6.03
   6.50   6.37
   7.00   6.05
   7.50   4.70
   8.00   1.90

C++ 版と同じ結果が得られました。

3. グラフ

参考までに、 R でグラフを作成してみた。

---

以上。