Python编写给你任一堆木块的层数，求出这堆木块的数量案例程序

题目描述

在墙角堆放着一堆完全相同的正方体小木块，如下图所示： 因为木块堆得实在是太有规律了，你只要知道它的层数就可以计算所有木块的数量了。 现在请你写个程序 给你任一堆木块的层数，求出这堆木块的数量.

输入格式

第一行是一个整数N(N<=10)表示测试数据的组数） 接下来的n行 每行只有一个整数 ，表示这堆小木块的层数

输出格式

对应每个输入的层数有一个输出，表示这堆小木块的总数量，每个输出占一行

输入样例

2

1

5

输出样例

1

35

源代码

T = int(input())  # 测试数据组数

for i in range(T):
    n = int(input())  # 小木块的层数
    total = n * (n + 1) // 2  # 小木块的总数量
    print(total)

首先读入测试数据的组数T，然后使用range函数循环T次，每次读入一个小木块的层数n。

计算小木块的总数量时，我们可以利用等差数列求和公式：

因为每一层的小木块数量是递增的，所以首项为1，末项为$n$，项数也为$n$，代入公式即可得到小木块的总数量。

最后将结果输出即可。

希望这个解题思路能够帮助到你！