2050 折线分割平面

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