파이썬/코딩테스트

[백준] 한수 구하기 - Python(파이썬)

대전집주인 2024. 3. 23. 16:18
728x90
SMALL

브루트포스 알고리즘으로

전체를 다 탐색하면서 결과를 구하는 알고리즘이다.

이문제의 경우 자리수의 합이 등차수열을 이루고 있으면 한수라고 나타내는데

1부터 99의 경우의 모든 수는 등차수열을 이루고 있고 100 이상 1000이하의 수는

각 자리수의 합을 비교함으로 한수의 개수를 구하는 프로그램을 구현하였다.

 

n = int(input())

if n < 100:
    print(n)
else:
    cnt = 0
    for i in range(100, n + 1):
        result = list(map(int, str(i)))
        a = result[0] - result[1]
        b = result[1] - result[2]

        if a == b:
            cnt += 1
    print(cnt + 99)
728x90
LIST