[백준] 1065번_한수

문제 링크 : https://www.acmicpc.net/problem/1065

1065번: 한수

n = int(input())

arr = [0] * (n+1)

for i in range(len(arr)):
    if 1<= i <= 99:
        arr[i] = 1
    elif 100<= i <= 999:
        hund = i // 100
        ten = i % 100 //10
        one = i % 100 % 10
        if (hund-ten) == (ten-one):
            arr[i] = 1

print(sum(arr))

처음에 떠오른건 배열로 만든거여서 이렇게 했지만 더 간단하게...

n = int(input())
count = 0
for i in range(1,n+1):
    if i<=99:
        count += 1
    elif 100<=i<=999:
        i = str(i)
        if (int(i[0])-int(i[1])) == (int(i[1])-int(i[2])):
            count+=1
print(count)

아니면

i = str(i)
        if (int(i[0])-int(i[1])) == (int(i[1])-int(i[2])):
            count+=1

이 부분 대신

new_i = list(map(int,str(i)))
	if (new_i[0]-new_i[1]) == (new_i[1]-new_i[2]):

로 바꿔줘도 된다. list(map)을 통해 자릿수 별로 떼어내어 리스트에 저장해주기 때문이다.