알고리즘 문제/CodeWar

Find the next perfect square!

BEstyle 2022. 10. 5. 13:02

DESCRIPTION:

You might know some pretty large perfect squares. But what about the NEXT one?

Complete the findNextSquare method that finds the next integral perfect square after the one passed as a parameter. Recall that an integral perfect square is an integer n such that sqrt(n) is also an integer.

If the parameter is itself not a perfect square then -1 should be returned. You may assume the parameter is non-negative.

Examples:(Input --> Output)

121 --> 144
625 --> 676
114 --> -1 since 114 is not a perfect square

def find_next_square(sq):
    num=sq**(1/2)
    num2=round(num)
    if num!=num2:
        return -1
    return (num2+1)**2