파이썬으로 이차 프로그래밍 문제 풀기

14 Nov 2023

python

이차 방정식은 다음과 같은 형태로 나타낼 수 있다.

$ax^2 + bx + c = 0$

여기서 a, b, c는 실수이고, x는 변수를 나타낸다.

파이썬에서 이차 방정식을 풀기 위해서는 다음과 같은 코드를 사용할 수 있다.

import math

def solve_quadratic_equation(a, b, c):
    discriminant = (b ** 2) - (4 * a * c)
    
    if discriminant > 0:
        x1 = (-b + math.sqrt(discriminant)) / (2 * a)
        x2 = (-b - math.sqrt(discriminant)) / (2 * a)
        return x1, x2
    elif discriminant == 0:
        x = -b / (2 * a)
        return x
    else:
        return "No real roots"

# 예시 방정식을 풀어보자
a = 1
b = -3
c = 2

solution = solve_quadratic_equation(a, b, c)

if isinstance(solution, tuple):
    print(f"The solutions are {solution[0]} and {solution[1]}")
else:
    print(f"The solution is {solution}")

위 코드에서는 solve_quadratic_equation 함수를 이용하여 이차 방정식을 풀고 있습니다. 이 함수는 이차 방정식의 계수 a, b, c를 입력으로 받아서 다음과 같은 결과를 반환합니다.

만약 방정식에 실근이 두 개인 경우, (x1, x2) 형태의 튜플로 반환합니다.
만약 방정식에 중근이 있는 경우, 실근 x를 반환합니다.
만약 방정식에 실근이 없는 경우, "No real roots"를 반환합니다.

위 코드에서는 예시로 a = 1, b = -3, c = 2라는 방정식을 풀고 있습니다. 이 경우에는 두 개의 실근이 존재하므로 "The solutions are 2.0 and 1.0"이라는 결과가 출력됩니다.

이처럼 파이썬을 이용하여 이차 방정식을 풀 수 있으며, math 모듈의 sqrt 함수를 통해 제곱근을 계산할 수 있습니다.

#수학 #알고리즘