Chương trình Python tìm tổng dãy số với số hạng thứ n là n^2 – (n-1)^2

Trong bài viết này, chúng ta sẽ học về giải pháp cho bài toán cho trước:

Đề bài

Chúng ta được cho một số nguyên đầu vào n và cần tính tổng của tất cả n số hạng nơi số hạng thứ n trong dãy được biểu diễn như sau:

Tn = n^2 - (n-1)^2

Chúng ta có công thức trực tiếp để tính tổng bao gồm phép nhân bình phương của n liên quan đến độ phức tạp thời gian hơn. Để giảm thiểu đó, chúng ta sử dụng phương pháp nhân mô-đun (modular multiplication) ở đây.

Bây giờ hãy xem triển khai:

Ví dụ

# Python program to find sum of given series.
mod = 1000000007
def findSum(n):
    return ((n % mod) * (n % mod)) % mod
# main()
n = 229137999
print(findSum(n))

Kết quả

218194447

Tất cả biến được khai báo trong khung toàn cục như hình dưới đây:

Kết luận

Trong bài viết này, chúng ta đã học về cách tìm tổng dãy số với số hạng thứ n là n^2 – (n-1)^2.