taum_and_bday.py

# source: https://www.hackerrank.com/challenges/taum-and-bday/problem
# video: https://youtu.be/9vajRE0bJ5w

def taumBday(b, w, bc, wc, z):
    # if black gifts > white gifts + z conversion cost
    if bc > wc + z:
        return (w * wc) + (b * (wc + z))
    # if white gifts > black gifts + z conversion cost
    if wc > bc + z:
        return (b * bc) + (w * (bc + z))
    # else, just buy as is
    return b * bc + w * wc