Расчет площади фигур

Жители страны Малевии часто экспериментируют с планировкой комнат. Комнаты бывают треугольные, прямоугольные и круглые. Чтобы быстро вычислять жилплощадь, требуется написать программу, на вход которой подаётся тип фигуры комнаты и соответствующие параметры, которая бы выводила площадь получившейся комнаты.
Для числа π в стране Малевии используют значение 3.14.

Решение:

fig = input()
if fig == "треугольник":
 a = float(input())
 b = float(input())
 c = float(input())
 p = (a + b + c) / 2
 s = (p * (p - a) * (p - b) * (p - c)) ** 0.5 
 print (s)
elif fig == "прямоугольник":
 a = float(input())
 b = float(input())
 s = a * b
 print (s)
elif fig == "круг":
 r = float(input())
 s = 3.14 * r**2
 print (s)