数数小木块最优代码的简便解法

做出来了，但是最优代码要简便的多

数数小木块

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

难度： 1

描述

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

 

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

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

输入

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

输出

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

样例输入

2
1
5

样例输出

1
35

最优代码：

关键：

n*(n+1)*(n+2)/6；

 
#include <stdio.h>
int main()
{
	int n,t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		printf("%d\n",n*(n+1)*(n+2)/6);
	}
}        

你的解法：

#include <bits/stdc++.h>
using namespace std;
int JS(int a)
{
    int i,s=0;
    for(i=a;i>=1;i--)
        s=s+i;
    return s;
}
int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        int a,j,sum=0;
        cin>>a;
        for(j=1;j<=a;j++)
            sum=sum+JS(j);
        cout<<sum<<endl;
    }
    return 0;
}