TIL_8주차(2) 코드카타: 알고리즘(Python)

알고리즘
1. 부족한 금액 계산하기

풀이:

def solution(price, money, count):
    for i in range(1, count+1):
        money -= price*i
    if money < 0:
        money =abs(money)
    else:
        money = 0
    return money

2. 문자열 다루기 기본

풀이:

def solution(s):
    if len(s) == 4 or len(s) == 6:
        if s.isdigit() == True:
            return True
        else:
            return False
    else:
        return False
    
    print(solution('a234'))
    print(solution('1234'))