Погрешность вычислений (точность вычисления дискриминанта)

levelp · May 23, 2015 · 14bff05 · 14bff05
1 parent 9a243cb
commit 14bff05
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/main/java/QuadraticEquation.java b/src/main/java/QuadraticEquation.java
@@ -2,6 +2,7 @@
  * Квадратное уравнение
  */
 public class QuadraticEquation {
+    public static final double EPS = 1e-10;
 
     /**
      * Решение квадратного уравнения: ax^2 + bx + c = 0
@@ -14,7 +15,7 @@ public class QuadraticEquation {
     public static double[] solve(double a, double b, double c) {
         // Дискриминант
         double D = Math.pow(b, 2) - 4 * a * c;
-        if (D == 0)
+        if (Math.abs(D) < EPS)
             return new double[]{-b / (2 * a)};
         double d = Math.sqrt(D);
         return new double[]{(-b - d) / (2 * a), (-b + d) / (2 * a)};

diff --git a/src/test/java/QuadraticEquationTest.java b/src/test/java/QuadraticEquationTest.java
@@ -24,4 +24,15 @@ public void testSimpleOneRoot() {
     public void testTwoSolutions() {
         assertArrayEquals("(x-1)(x-2) = x^2-3x+2", new double[]{1, 2}, QuadraticEquation.solve(1, -3, 2), EPS);
     }
+
+    /**
+     * Погрешность вычислений (точность вычисления дискриминанта)
+     */
+    @Test
+    public void testOneRootAccuracy() {
+        for (int q = 1; q < 999; q++) {
+            double t = 0.3;
+            assertArrayEquals("t(x-q)^2 = t*x^2 - 2*t*q*x + t*q^2", new double[]{q}, QuadraticEquation.solve(t, -2 * t * q, t * q * q), EPS);
+        }
+    }
 }