[HackerRank] Library Fine

[문제 링크]

[입력]

d1, m1, y1: 반납하는 일,월,년도 , 모두 정수
d2, m2, y2: 예정된 반납 일,월 , 년도 , 모두 정수

[출력]

만약 예정된 반납일 이전 또는 반납일에 제출하면 지불 금액 없음
만약 예정된 반납일 보다 늦지만 같은 달에 제출하면 "15(d2-d1)" 만큼 지불
만약 예정된 반납일 보다 늦지만 같은 년도에 제출하면 "500(m2-m1)" 만큼 지불
만약 예정된 년도보다 늦게 제출하면 10000 지불

[코드]

def libraryFine(d1, m1, y1, d2, m2, y2):
    # Write your code here
    if(y1<y2):
        return 0
    elif (y1==y2):
        if(m1<m2):
            return 0
        elif(m1==m2):
            if(d1<=d2):
                return 0
            else:
                return 15*(d1-d2)
        else:
            return 500*(m1-m2)
    else:
        return 10000