โปรแกรมหาผลรวมของ n เลขคี่แรกใน Python

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

เช่น ถ้าอินพุตคือ 7 ผลลัพธ์จะเป็น 49 เพราะ 1+3+5+7+9+11+13 = 49

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

• ถ้า n == 0 คืนค่า 0
• sum := 1, count := 0, temp := 1
• while count < n-1 ทำ
  • temp := temp + 2
  • sum := sum + temp
  • count := count + 1
• คืนค่า sum

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

class Solution:
    def solve(self, n):
        if n == 0:
            return 0
        sum = 1
        count = 0
        temp = 1
        while count < n-1:
            temp += 2
            sum += temp
            count += 1
        return sum

ob = Solution()
print(ob.solve(7))

อินพุต

7

เอาต์พุต

49