HDU-2050 折线分割平面

#include <bits/stdc++.h>
#define pb push_back
#define _for(i,a,b) for(int i = (a);i < (b);i ++)
#define INF 0x3f3f3f3f

using namespace std;

const int maxn = 5003;

int main()
{
    int T;
    scanf("%d",&T);
    int N;
    while(T--)
    {
        scanf("%d",&N);
        cout << 2*N*N-N+1 << endl;
    }
    return 0; 
}