• + 1 comment

    Similarly:

    print(max(height) - k if max(height) > k else 0)
    

    is what I did