Python 백준 알고리즘 : 1065 한수

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

1065번: 한수

 

# 1065 한수

def hansu(num) :
  hansu_cnt = 0
  for i in range(1, num+1) :
    num_list = list(map(int, str(i)))
    if i < 100 :
      hansu_cnt += 1
    elif num_list[0] - num_list[1] == num_list[1] - num_list[2] :
      hansu_cnt += 1
  return hansu_cnt

x = int(input())
print(hansu(x))

 

함수를 이용했다.

 

x값을 입력받아서 hansu() 함수에 대입한다. 범위 안에 한수의 개수를 저장할 변수 hansu_cnt를 초기화한다. num값을 포함하는 범위이기에 for문에 범위를 1~num+1로 두었다. 변수 num_list에 1부터 문자열로 변환하여 저장한다.

 

100 미만은 모두 한수이기에 100 미만인 수일 때는 hansu_cnt 값에 1을 더해주고, 그 이상의 값들은 등차가 같은지 확인하기 위해서 num_list[0] - num_list[1] 값과 num_list[1] - num_list[2] 값이 같으면 hansu_cnt 값에 1을 더해준다.

 

여기서 그 다음 자릿수를 확인하지 않는 이유는 문제에서 1,000 이하의 자연수라고 조건을 줬기 때문이다.