Что не так в коде (Python3)?

Question

[Iqv]

Есть задача:
Вводится строка в формате
ax + b = 0
или
ax - b = 0
где вместо a — целое число, вместо b — целое неотрицательное число. Числа по модулю не превосходят 1000. Число a может быть пропущено, если оно равно 1. b всегда присутствует, x также всегда присутствует, даже если а = 0.
Выведите корни уравнения, если их конечное число; 'NO', если корней нет и 'INF', если корней бесконечно много.
Моё решение:

s = input()
a = s[:s.index("x")]
if s.find('+') != -1:
    b = -int(s[s.index("+")+1:s.index("=")])
else:
    b = int(s[s.index("-")+1:s.index("=")])
if a == "0" and b == 0:
    print("INF")
elif a == "0" and b != 0:
    print("NO")
elif a == "":
    print(b)
else:
    print(b/int(a))

Partial Solution. Your score is = 9, 9/10
Что не так с кодом?

Comments

[longclaps]

Это что за проверяющая система?

[Iqv]

Домашнее задание на сайте foxford.ru

[longclaps]

Iqv:
не хочу там регистрироваться, может, им формат вывода не нравится. попробуй прогнать

s = " 3 x + 3 = 0"
left, _ = s.replace(' ', '').split('=')
a, b = left.split('x')
a, b = float(a) if a else 1., float(b)
if a:
x = -b / a
print(int(x) if x.is_integer() else x)
else:
print("NO" if b else "INF")

[Iqv]

10/10 tests passed, большое спасибо!

Answer 1

[Sly_tom_cat .]

Используйте лучше регулярные выражения:

s = input()
res = re.findall(r'(-?\d?)x([+-]\d)=0', s)[0]
a = (-1 if res[0] == '-' else
     1 if not res[0] else
     int(res[0]))
b = int(res[1]) 
print(('INF' if b == 0 else 'NO') if a == 0 else -b/a)

Comments

[Iqv]

res = re.findall(r'(-?\d?)x([+-]\d)=0', s)[0] Можете пояснить что это?

[Sly_tom_cat .]

re.findall ищет все совпадения регулярного выражения(1-й параметр) в строке(второй параметр). Нас интересует только первое совпидение поэтому сразу вытаскиваем 0-вой элемент [0].
Регулярка простая : минус может встречаться 0или1 раз(такое число повторений обозначается как ?), цифра (\d) может встречаться 0/1 раз, вместетони дают а(скобочками выделен первый возвращаемый элемент). Далее долже быть икс. Дальше либо +, либо -(выбор одного из несольких - []), потом должна быть цифра. Знак +/- и цифра дают второй возаращаемый элемент.
Из первого делаем а, из второго b.

[Iqv]

Спасибо!

Answer 2

[Sayonji]

У вас не учитываются отрицательные а: не пройдёт тест -5x-5=0, например.