h215 - 客製金莎巧克力金字塔

🔗 前往 ZeroJudge 原題

題目描述

題目要求計算指定層數的金字塔形巧克力所需的總巧克力數量。金字塔的每一層都包含該層數的巧克力，例如第一層有 1 個，第二層有 2 個，以此類推。

解題思路

此題為簡單的數學問題。金字塔的總巧克力數量等於從 1 到 n 的所有整數的平方和。程式碼直接使用迴圈計算這個總和。由於 n 的最大值為 1023，迴圈執行次數有限，因此效率足夠高。

複雜度分析

• 時間複雜度: O(n)
• 空間複雜度: O(1)

程式碼

#include <bits/stdc++.h>
using namespace std;
long long ans,n;
int main(){
	cin >> n;
	for(long long i=1;i<=n;++i)
		ans+=i*i;
	cout << ans;
}