[백준] 1065번 한수 - 파이썬

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

1065번: 한수

풀이 1

주어진 자연수가 자릿수 2 이하이면 등차수열 조건을 만족한다.

자릿수가 3 이상인 경우에는 자릿수가 1 차이나는 숫자끼리 차이가 같은지 확인한다.

# 1065번 한수
n = int(input())
cnt = 0

def check(standard):
    for j in range(1, len(word)-1):
            if (int(word[j]) - int(word[j+1])) != standard:
                return False
    return True

for i in range(1, n+1):
    word = str(i)
    if len(word) <= 2:
        cnt += 1
    else:
        standard = int(word[0]) - int(word[1])
        if check(standard):
            cnt += 1
                
print(cnt)

출력결과

 

풀이 2

입력값은 1000 이하의 자연수라는 조건이 주어진 것을 이용해도 좋다.

# 1065번 한수
n = int(input())
cnt = 0

def check(i):
    word = list(map(int, str(i)))
    if 2*word[1] != word[0] + word[2]:
        return False
    return True

if n <= 99:
    cnt = n
else:
    cnt = 99
    for i in range(100, n+1):
        if check(i):
            cnt += 1
print(cnt)

출력결과