[COS Pro 1급, Python] 3차 7번 : 카프리카 수

문제 유형

 한줄 수정

난이도

 easy

Note 

 1. 카프리카 수 : 자신의 제곱수를 둘로 나눠 더한 값이 자기 자신 ( 단, 둘로 나뉜 수는 양수 )

  ex ) 55 -> 55^2 = 3025 -> 30+25 = 55

 2. 코딩 유형으로 나온다면, divisor를 사용한다는 것만 떠올리면 될 듯.

 

Code

# -*- coding: utf-8 -*-
# UTF-8 encoding when using korean

def solution(k):
	answer = []
	for i in range(1, k + 1):
		square_num = i * i
		divisor = 1
		while square_num // divisor != 0:
			front = square_num // divisor
			back = square_num % divisor
			divisor *= 10
			if back != 0 and front != 0:
				if front + back == i:
					answer.append(i)
	return answer

k = 500
ret = solution(k)

print("solution 함수의 반환 값은", ret, "입니다.")

 

※ 가끔 코드 중 print(~)가 있습니다. 정리 못한 점 죄송합니다.