โปรแกรมหาผลรวมของจำนวนคี่ N ตัวแรกใน Python

สมมติว่าเรามีตัวเลข n เราต้องหาผลรวมของจำนวนคี่บวก n ตัวแรก

เช่น ถ้าอินพุตคือ n = 10 ผลลัพธ์จะเป็น 100 เพราะจำนวนคี่ 10 ตัวแรกคือ [1, 3, 5, 7, 9, 11, 13, 15, 17, 19] ผลรวมคือ 100

ขั้นตอนการแก้ปัญหา

1. มีข้อสังเกตที่ดี: ผลรวมของจำนวนคี่ n ตัวแรก เสมอเท่ากับ n^2 (กำลังสองของ n)
2. ดังนั้นเพื่อหาผลลัพธ์ ให้คืนค่า n * n

ตัวอย่างการใช้งาน

def solve(n):
    return n * n

n = 10
print(solve(n))

อินพุต

10

เอาต์พุต

100