Baekjoon 1065.py [한수]

문제가 궁금하다면?

내 풀이

import sys


def isHanNum(one, ten, hun):
    first = ten - one
    second = hun - ten
    if first == second:
        return 1
    return 0


def d(number):
    if 1 <= number <= 99:
        return 1
    elif 100 <= number <= 999:
        ones = number % 10
        tens = (number // 10) % 10
        huns = (number // 10) // 10
        return isHanNum(ones, tens, huns)
    else:
        return 0


input = sys.stdin.readline
n = int(input().rstrip())
cnt = 0
result = []

while n != cnt:
    result.append(d(cnt+1))
    cnt += 1

print(sum(result))

풀이 복기

• 문자열로 변환해 자리수 계산하는방법(이게 더 편해보임)

numbers = list(map(int, str(number)))
if numbers[2] - numbers[1] == numbers[1] - numbers[0]:
	return 1

• 각 수마다 한수 판별한 값을 리스트에 저장했는데, 그냥 변수에 저장하는게 더 좋았을듯함