HDU——2050——折线分割平面

#include <iostream>
using namespace std;
long long a[10002];
void fn()
{
a[0]=0;
a[1]=2;
a[2]=7;
for(int i=3;i<10002;i++)
{
a[i]=a[i-1]+4*(i-1)+1;
}
}
int main()
{
int n;
int b;
cin>>n;
fn();
for(int i=1;i<=n;i++)
{
cin>>b;
cout<<a[b]<<endl;
}
return 0;
}