Chương trình tìm tổng n số lẻ đầu tiên trong Python

Cho số n, tìm tổng của n số lẻ dương đầu tiên.

Ví dụ

Đầu vào: 7.
Đầu ra: 49 (1+3+5+7+9+11+13 = 49).

Thuật toán

1. Nếu n == 0 trả về 0.
2. Khởi tạo sum = 1, count = 0, temp = 1.
3. Khi count < n-1:
   • temp += 2, sum += temp, count += 1.
4. Trả về sum.

Mã nguồn Python

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))

Đầu vào / Đầu ra

Input: 7
Output: 49