[코딩 테스트 연습(파이썬/Python)] 백준 1065번 _ 한수

https://www.acmicpc.net/problem/1065

1065번: 한수

[문제]

어떤 양의 정수 X의 각 자리가 등차수열을 이룬다면, 그 수를 한수라고 한다. 등차수열은 연속된 두 개의 수의 차이가 일정한 수열을 말한다. N이 주어졌을 때, 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력하는 프로그램을 작성하시오. 

[입력]

첫째 줄에 1,000보다 작거나 같은 자연수 N이 주어진다.

[출력]

첫째 줄에 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력한다.

[아이디어]

1) 100 미만의 자연수는 모두 조건에 해당한다. 즉 100보다 작을 때는 입력을 그대로 출력하면 된다.

2) 1000은 해당하지 않으므로 세 자리 자연수만 신경쓰면 된다.

3) 세 자리 자연수는 가운데 숫자*3이 각 자리 숫자의 합이면 조건에 해당한다.

 

n = int(input())
if n < 100:
    print(n)
else:
    cnt = 99
    for i in range(100, n+1):
        if int(str(i)[1]) * 3 == sum(map(int, list(str(i)))):
            cnt += 1
    print(cnt)