[CodingTest] 백준 #1065 한수

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

1065번: 한수

 

 

내 코드

def check(num):
    n = str(num)
    arr =[]
    flag=0
    for i in range(len(n)-1):
        if len(arr)==0:
            arr.append(int(n[i+1])-int(n[i]))
        else:
            if arr[0]!=int(n[i+1])-int(n[i]):
                flag=1
                return False
            
    if flag==0:
        return True
    
n = int(input())
sum=0
for i in range(1,n+1):
    if check(i):
        sum+=1
print(sum)

숫자를 문자열로 변환하고, 만약 arr이 비어있다면 다음 자릿수에서 현재 자릿수를 뺀 값을 arr에 넣어준다.

arr이 있다면, arr에 들어있는 값과 자릿값들의 차를 비교한다.

만약 arr에 들어있는 값과 차가 다르면, False를 리턴하고, 끝까지 비교했을 때 모두 동일하다면 True를 리턴한다.