백준 1065 - 한수 (파이썬)

문제

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

입력

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

출력

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

 

 

# 해결법

 

우선 등차수열인 수를 담아줄 set()을 생성했다. n이 1~99 일때는 모두 등차수열이기때문에 전부 num에 담아줬다.

세자리수일경우는 100의 자리 - 10의 자리 == 10의 자리 - 1의 자리 일 경우에만 num에 담도록 했다.

마지막에 len() 을 이용하여 출력하였다. 

 

# 나의 코드 

n = int(input())
num = set()
for i in range(1,n+1):
    if i < 100:
        num.add(i)
    elif i < 1000:
        if (((i//100)-((i//10)%10)) == (((i//10)%10)-(i%10))) :
            num.add(i)
print(len(num))