หาระยะทางรวมในการเก็บหินที่วางห่างเท่ากันใน Python

สมมติว่ามีการแข่งขัน หินวางห่างกัน 4 หน่วย หน่วยแรกห่างจากถัง 6 หน่วย ผู้แข่งเริ่มจากถัง วิ่งเก็บหินใกล้สุด กลับวางถัง วิ่งเก็บหินถัดไป... จนครบ n ก้อน หาระยะทางรวม

เช่น n = 5 ผลลัพธ์คือ 140

สูตร: D = 2 × [6n + 2n(n-1)]

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

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

n = 5
print(find_distance(n))

อินพุต

5

เอาต์พุต

140