Finding roots of polynomial in Java

问题:

I need to find (approximate, numerical) solution to Legendre polynomial. I tried the several Java libraries but none have what I am looking for (the closest is commons-math which even has code for finding the solutions in Laguerre Solver but does not expose the method). Is there an existing solution or do I need to implement my own?

回答1:

You can use efficient-java-matrix-library

Please find the below sample example for the same

public class PolynomialRootFinder {

    /**
     * <p>
     * Given a set of polynomial coefficients, compute the roots of the polynomial.  Depending on
     * the polynomial being considered the roots may contain complex number.  When complex numbers are
     * present they will come in pairs of complex conjugates.
     * </p>
     *
     * @param coefficients Coefficients of the polynomial.
     * @return The roots of the polynomial
     */
    public static Complex64F[] findRoots(double... coefficients) {
        int N = coefficients.length-1;

        // Construct the companion matrix
        DenseMatrix64F c = new DenseMatrix64F(N,N);

        double a = coefficients[N];
        for( int i = 0; i < N; i++ ) {
            c.set(i,N-1,-coefficients[i]/a);
        }
        for( int i = 1; i < N; i++ ) {
            c.set(i,i-1,1);
        }

        // use generalized eigenvalue decomposition to find the roots
        EigenDecomposition<DenseMatrix64F> evd =  DecompositionFactory.eigGeneral(N, false);

        evd.decompose(c);

        Complex64F[] roots = new Complex64F[N];

        for( int i = 0; i < N; i++ ) {
            roots[i] = evd.getEigenvalue(i);
        }

        return roots;
    }
}

回答2:

Since release 3.1, Commons Math supports finding all complex roots of a polynomial function.

See LaguerreSolver#solveAllComplex

回答3:

commons math has reasonable API for polynomials

    //  -4 + 3 x + x^2
    PolynomialFunction polynomial = new PolynomialFunction(new double[]{ -4, 3, 1});
    LaguerreSolver laguerreSolver = new LaguerreSolver();
    double root = laguerreSolver.solve(100, polynomial, -100, 100);
    System.out.println("root = " + root);