[백준] 1065 파이썬 (한수)

코드

def hansu(num):
    count = 0
    for i in range(1, num+1):
        if i<100:
            count += 1
        else:
            num_list = list(map(int, str(i)))
            if num_list[1]-num_list[0] == num_list[2]-num_list[1]:
                count += 1
    return count

if __name__ == "__main__":
    n = int(input())
print(hansu(n))

풀이

1보다 크거나 같고, n보다 작거나 같은 한수의 개수를 구하는 함수는 다음과 같다.

def hansu(num):
    count = 0
    for i in range(1, num+1):
        if i<100:
            count += 1
        else:
            num_list = list(map(int, str(i)))
            if num_list[1]-num_list[0] == num_list[2]-num_list[1]:
                count += 1
    return count

먼저 한수의 개수를 0으로 초기화한다.
N보다 작은 수를 반복문을 통해 한수인지 검사한다.


100보다 작은 수 (ex. 99)는 무조건 한수이므로 count += 1 해준다.

그 외의 경우는 100보다 크거나 같고 1000보다는 작거나 같은 수인데

num_list[1]-num_list[0] == num_list[2]-num_list[1]

를 통해 등차수열인지 판별한다.

1000의 경우는 등차수열이 아니므로 고려하지 않아도 되므로 위 조건식으로 등차수열인지 판별하 수 있다.