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

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

 

알고리즘 분류: 브루트포스 알고리즘

 

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

1065번: 한수

 

문제

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

입력

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

출력

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

---

제출한 답

#a가 한수인지 아닌지 확인하는 함수
def cold(a):
  arr=list()
  
  for i in str(a):# a 문자전환 후 반복문
    arr.append(int(i))# 숫자화, 리스트에 담기
    
  # 해당 숫자의 끝과 중간값 비교
  num = arr[0]+arr[len(arr)-1]
  for i in range(1, int(len(arr)/2)+1):
    if num != arr[i]+arr[len(arr)-i-1]:
      return False
  return True
  
###

n = int(input())
count = 0
for i in range(1,n+1):
  if cold(i):
    count += 1
print(count)

더 쉬운 답안도 있지만

1000 이상의 자연수 N에도 적용되는 함수가 

범용성이 좋을 것 같아 이렇게 짜봤다.