Tìm khoảng cách đi thu thập vật phẩm cách đều nhau trong Python

Cuộc đua: xô ở điểm xuất phát, cách đáy 1 là 6 đơn vị. Các đáy cách nhau 4 đơn vị. Người chơi từ xô chạy lấy đáy 1 về, rồi đáy 2, ... đến đáy n. Tính tổng khoảng cách chạy.

Ví dụ: n=5 → 140 (2*6 + 2*(6+4) + 2*(6+8) + 2*(6+12) + 2*(6+16) = 140).

Giải thuật (Công thức toán học)

Khoảng cách cho đáy i: 2 * (6 + 4*(i-1)). Tổng:

D = 2 * [6n + 4 * (1+2+...+(n-1))] = 2 * [6n + 2n(n-1)]

Ví dụ

def find_distance(n):
    return 2 * (6 * n + 2 * (n - 1) * n)

n = 5
print(find_distance(n))

Đầu vào

5

Đầu ra

140