Question

what i'm trying to do is write a quadratic equation solver but when the solution should be -1, as in quadratic(2, 4, 2) it returns 1

what am i doing wrong?

#!/usr/bin/python
import math
def quadratic(a, b, c):
        #a = raw_input("What\'s your `a` value?\t")
        #b = raw_input("What\'s your `b` value?\t")
        #c = raw_input("What\'s your `c` value?\t")
        a, b, c = float(a), float(b), float(c)
        disc = (b*b)-(4*a*c)
        print "Discriminant is:\n" + str(disc)
        if disc >= 0:
                root = math.sqrt(disc)
                top1 = b + root
                top2 = b - root
                sol1 = top1/(2*a)
                sol2 = top2/(2*a)
                if sol1 != sol2:
                        print "Solution 1:\n" + str(sol1) + "\nSolution 2:\n" + str(sol2)
                if sol1 == sol2:
                        print "One solution:\n" + str(sol1)
        else:
                print "No solution!"

EDIT: it returns the following...

>>> import mathmodules
>>> mathmodules.quadratic(2, 4, 2)
Discriminant is:
0.0
One solution:
1.0
Was it helpful?

Solution

Unless the formula has changed since I went to school (one can never be too sure), it's (-b +- sqrt(b^2-4ac)) / 2a, you have b in your code.

[edit] May I suggest a refactor?

def quadratic(a, b, c):
    discriminant = b**2 - 4*a*c
    if discriminant < 0:
      return []
    elif discriminant == 0:
      return [-b / (2*a)]
    else:
      root = math.sqrt(discriminant)
      return [(-b + root) / (2*a), (-b - root) / (2*a)]

print quadratic(2, 3, 2) # []
print quadratic(2, 4, 2) # [-1]                    
print quadratic(2, 5, 2) # [-0.5, -2.0]

OTHER TIPS

The solution to the quadratic is

x = (-b +/- sqrt(b^2 - 4ac))/2a

but what you have coded up is

x = (b +/- sqrt(b^2 - 4ac))/2a

So that's why you get the sign error.

The signs of top1 and top2 are wrong, see http://en.wikipedia.org/wiki/Quadratic_equation

top1 = b + root
top2 = b - root

Should be:

top1 = -b + root
top2 = -b - root
Licensed under: CC-BY-SA with attribution
Not affiliated with StackOverflow
scroll top