ตรวจสอบว่าเลขบวก 1 แล้วเป็น Perfect Square ได้หรือไม่ด้วย Python

โจทย์: กำหนดเลข n ตรวจสอบว่า n+1 เป็น Perfect Square หรือไม่

ตัวอย่าง

n = 288 → 288+1=289=17² → True

อัลกอริทึม

1. res_num = n + 1
2. sqrt_val = int(sqrt(res_num))
3. ตรวจสอบ sqrt_val² == res_num

โค้ดตัวอย่าง

from math import sqrt

def solve(n):
    res_num = n + 1
    sqrt_val = int(sqrt(res_num))
    if sqrt_val * sqrt_val == res_num:
        return True
    return False

n = 288
print(solve(n))

Input / Output

Input: 288
Output: True