NYOJ-ACM-数数小木块

数数小木块

时间限制：3000 ms  |  内存限制：65535 KB
难度：1

描述

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



 

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

现在请你写个程序 给你任一堆木块的层数，求出这堆木块的数量.

输入第一行是一个整数N(N<=10)表示测试数据的组数）

接下来的n行 每行只有一个整数 ，表示这堆小木块的层数，
输出对应每个输入的层数有一个输出，表示这堆小木块的总数量，每个输出占一行
样例输入

2
1
5

样例输出

1
35

思路：

这道题只要知道他的规律是n*(n+1)*(n+2)/6就能做出来了

代码：

import java.util.Scanner;

public class Main {
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner sc = new Scanner(System.in);
int a = sc.nextInt();
int[] b = new int[a];//输入数组
int[] sum = new int[a];//结果数组
for(int i=0;i<a;i++){
b[i] = sc.nextInt();
}
for(int i=0;i<a;i++){
sum[i] = b[i]*(b[i]+1)*(b[i]+2)/6;//木块规律 n*(n+1)*(n+2)/6
}
for(int i=0;i<a;i++){//输出结果
System.out.println(sum[i]);
}

}
}