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]);
		}
		
	}
}