3-7_KaprekarNumber.py

# -*- 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, "입니다.")