are_they_the_same_6kyu.py

"""
comp(a, b) returns true because in b 121 is the square of 11,
14641 is the square of 121,
20736 the square of 144,
361 the square of 19,
25921 the square of 161, and so on.

It gets obvious if we write b's elements in terms of squares:

a = [121, 144, 19, 161, 19, 144, 19, 11] 
b = [11*11, 121*121, 144*144, 19*19, 161*161, 19*19, 144*144, 19*19]

Invalid arrays

If we change the first number to something else, comp may not return true anymore:

a = [121, 144, 19, 161, 19, 144, 19, 11]  
b = [132, 14641, 20736, 361, 25921, 361, 20736, 361]

comp(a,b) returns false because in b 132 is not the square of any number of a.

a = [121, 144, 19, 161, 19, 144, 19, 11]  
b = [121, 14641, 20736, 36100, 25921, 361, 20736, 361]

comp(a,b) returns false because in b 36100 is not the square of any number of a.
Remarks

a or b might be [] (all languages except R, Shell).
a or b might be nil or null or None or nothing (except in Haskell, Elixir, C++, Rust, R, Shell).

If a or b are nil (or null or None), the problem doesn't make sense so return false.

If a or b are empty the result is evident by itself.
"""


def comp(array1, array2):
    result = len(array1) > 0
    
    if array1 == array2 == []:
        result = True
    
    for n in array1:
        squared = n * n
        
        verified = False
        
        if squared in array2:
            verified = True
            array2.remove(squared)
            
        result = result and verified
        
    return result