2017年4月1日土曜日

Java BigDecimalを用いたルート(平方根)の計算

このサイトでこのコードのテストができます。
https://www.jdoodle.com/online-java-compiler/

もともと書いてあるコードを全部消して下のコードをそのまま丸ごと張り付けし、「Execute」というボタンを押して実行してください。

実行結果

なおJava9だとBigDecimalにもsqrtメソッドがあるようです。
以下はJava8を前提にしています。

import java.math.BigDecimal;
import java.math.MathContext;

public class HelloWorld
{
   public static void main(String[] args)
   {
        BigDecimal sqrttest1 = sqrt(new BigDecimal("4"), 50); //Square root of 4 (50 significant digits)
        System.out.println(sqrttest1);
        BigDecimal sqrttest2 = sqrt(new BigDecimal("2"), 50); //Square root of 2 (50 significant digits)
        System.out.println(sqrttest2);
        BigDecimal sqrttest3 = sqrt(new BigDecimal("1"), 50); //Square root of 1 (50 significant digits)
        System.out.println(sqrttest3);
    }

    public static BigDecimal sqrt(BigDecimal a, int scale){

        BigDecimal x = new BigDecimal(Math.sqrt(a.doubleValue()), MathContext.DECIMAL64);
        if(scale < 17){
             x = x.setScale(scale, BigDecimal.ROUND_HALF_EVEN);
             return x;
        }

        BigDecimal b2 = new BigDecimal(2);
        for(int tempScale = 16; tempScale < scale; tempScale *= 2){
            x = x.subtract(
                    x.multiply(x).subtract(a).divide(
                    x.multiply(b2), scale, BigDecimal.ROUND_HALF_EVEN));
        }
        return x;
    }
}