自定义函数求一元二次方程

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        double a = input.nextDouble();
        double b = input.nextDouble();
        double c = input.nextDouble();
       
        double discriminant = b * b - 4 * a * c;
       
        if (discriminant > 0) {
            double x1 = (-b + Math.sqrt(discriminant)) / (2 * a);
            double x2 = (-b - Math.sqrt(discriminant)) / (2 * a);
            System.out.printf("x1=%.3f x2=%.3f%n", x1, x2);
        } else if (discriminant == 0) {
            double x1 = -b / (2 * a);
            System.out.printf("x1=%.3f%n", x1);
        } else {
            double realPart = -b / (2 * a);
            double imaginaryPart = Math.sqrt(-discriminant) / (2 * a);
            System.out.printf("x1=%.3f+%.3fi x2=%.3f-%.3fi%n", realPart, imaginaryPart, realPart, imaginaryPart);
        }
    }
}