Execicio 2 - Prova Python - Eng.Comp 26

# coding: utf-8

numPrimeiroAndar = int(input("Total de pessoas no primeiro andar: "))
numSegundoAndar = int(input("Total de pessoas no segundo andar: "))
numTerceiroAndar = int(input("Total de pessoas no terceiro andar: "))

totalDePessoas = numPrimeiroAndar + numSegundoAndar + numTerceiroAndar

##Caso a cafeteira fique no primeiro andar.
##Calculo o produto das pessoas pelos segundos que cada andar leva ate chegar a cafeteira.
##Temos entao pessoas*segundos
primeiro_custoPrimeiro = numPrimeiroAndar * 0
primeiro_custoSegundo = numSegundoAndar * 2
primeiro_custoTerceiro = numTerceiroAndar * 4
##Custo total primeiro andar // Media aritimetica entre os custos
##Calculamos o quociente do total dos custos pelo total de pessoas
##Temos entao (pessoas * segundo) / pessoas, isto nos da os segundos medios
##que uma pessoa em qualquer andar leva ate chegar a cafeteira
primeiro_custo = (primeiro_custoPrimeiro + primeiro_custoSegundo + primeiro_custoTerceiro) / totalDePessoas
###############

##Calculamos para dos dois outros casos

##Caso a cafeteira fique no segundo andar
segundo_custoPrimeiro = numPrimeiroAndar * 2
segundo_custoSegundo = numSegundoAndar * 0
segundo_custoTerceiro = numTerceiroAndar * 2
##Custo total primeiro andar // Media aritimetica entre os custos
segundo_custo = (segundo_custoPrimeiro + segundo_custoSegundo + segundo_custoTerceiro) / totalDePessoas
###############


##Caso a cafeteira fique no terceiro andar
terceiro_custoPrimeiro = numPrimeiroAndar * 4
terceiro_custoSegundo = numSegundoAndar * 2
terceiro_custoTerceiro = numTerceiroAndar * 0
##Custo total primeiro andar // Media aritimetica entre os custos
terceiro_custo = (terceiro_custoPrimeiro + terceiro_custoSegundo + terceiro_custoTerceiro) / totalDePessoas
###############

##Mostramos os resultados
print("-----------------------------")
print("Custo no primeiro andar: {:.5f}s".format(primeiro_custo))
print("Custo no segundo andar: {:.5f}s".format(segundo_custo))
print("Custo no terceiro andar: {:.5f}s".format(terceiro_custo))
print("-----------------------------")

##Escolhemos o menor dos custos
if (primeiro_custo < segundo_custo):
    if (primeiro_custo < terceiro_custo):
        print("A cafeteira deve ficar no Primeiro andar")
    else:
        print("A cafeteira deve ficar no Terceiro andar")
else:
    if (segundo_custo < terceiro_custo):
        print("A cafeteira deve ficar no Segungo andar")
    else:
        print("A cafeteira deve ficar no Terceiro andar")