[python]Bouncing Balls

https://www.codewars.com/kata/5544c7a5cb454edb3c000047/train/python

Codewars: Train your coding skills

 

한 아이가 h 높이에서 공을 창문으로 던질 때 공은 h 곱하기 bounce만큼 튀어 오른다. 아이의 엄마가 window에서 볼 때 엄마는 공이 지나가는 것을 몇 번이나 보는지가 문제이다.

 

 

변수 h, bounce와 window가 각 조건을 만족하지 않으면 -1을 반환해야 한다는 규칙이 있다. 

이 규칙을 잘 지키면서 h가 window보다 클 경우 내려갈 때와 올라갈 때를 구분지어서 코드를 작성하였다.

 

# My Code
def bouncingBall(h, bounce, window):
    cnt = 0
    if 0 < h and 0 < bounce < 1 and window < h:
        while h > window:
            # 내려갈 때
            cnt += 1
            h *= bounce
            if h > window:
                # 올라갈 때
                cnt += 1
        return cnt
    else:
        return -1
 
if __name__=='__main__':
    answer = bouncingBall(3, 0.66, 1.5)
    print(answer)
    answer = bouncingBall(30, 0.66, 1.5)
    print(answer)