【入门】数数小木块

养成好习惯，先赞后看！

题目描述

在墙角堆放着一堆完全相同的正方体小木块，如下图所示：

 

因为木块堆得实在是太有规律了，你只要知道它的层数就可以计算所有木块的数量了。

输入

只有一个整数 n ，表示这堆小木块的层数，已知1 <= n <= 100 。

输出

只有一个整数，表示这堆小木块的总数量。

样例输入

复制

5

样例输出

复制

35

来源/分类

第12章_自定义函数

上代码~

#include <bits/stdc++.h>
using namespace std;
int a(int t);
int b(int t);

int main() {
    int n;
    cin >> n;
    cout << b(n);

    return 0;
}

int a(int t) {
    return t * (t + 1) / 2;
}

int b(int t) {
    int ans = 0;
    for (int i = t; i; i--) {
        ans += a(i);
    }
    return ans;
}