Modified_Kaprekar_Numbers.py

#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the kaprekarNumbers function below.
def check(i):
    sq = str(i**2)
    le = len(str(i))
    r = sq[-le:]
    l = sq[:-le] or '0'
    return sum(map(int,(l,r)))==i

def kaprekarNumbers(p, q):
    return [i for i in range(p,q+1) if check(i)]

if __name__ == "__main__":
    p = int(input())
    q = int(input())
    print(*kaprekarNumbers(p, q) or ["INVALID RANGE"])