TDT4109/Exercise 2/11 - Andregradsligning/task11a.py

def getValues() -> (int, int, int):
  while True:
    values = input('Gi inn en andregradsliknings a, b og c separert med mellomrom:\n\t')
    try:
      splitValues = values.split(' ')
      assert len(splitValues) == 3
      return map(int, splitValues)
      
    except ValueError:
      print('Sørg for at alle tallene er heltall.\n')
    except AssertionError:
      print('Det skal bare være 3 tall.\n')

if __name__ == "__main__":
  a, b, c = getValues()
  d = b**2  - 4 * a * c

  if d > 0:
    print('Ligninga har to reelle løsninger')
  elif d == 0:
    print('Ligninga har en reell løsning')
  else:
    print('Ligninga har to imaginære løsninger')