โปรแกรมตรวจสอบจัดเรียง Array ให้ Diff คู่ติดกันเท่ากันได้หรือไม่ใน Python

ให้ลิสต์ nums ตรวจสอบว่าจัดเรียงใหม่ได้หรือไม่ ที่ Diff ระหว่างคู่ติดกันเท่ากันทุกคู่ (Arithmetic Progression)

ตัวอย่าง: nums=[8,2,6,4] → True (เรียง [2,4,6,8] Diff=2)

ขั้นตอน (Sort + Check)

1. N ≤ 2: True
2. เรียง nums
3. targetDiff = nums[1]-nums[0]
4. เช็ค nums[i]-nums[i-1] == targetDiff ทุก i

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

def solve(nums):
    N = len(nums)
    if N <= 2:
        return True
    nums.sort()
    targetDiff = nums[1] - nums[0]
    for i in range(2, N):
        if nums[i] - nums[i-1] != targetDiff:
            return False
    return True

nums = [8, 2, 6, 4]
print(solve(nums))  # Output: True

อินพุต

[8, 2, 6, 4]

เอาต์พุต

True