问题 A: 数格子

第二天叫醒我的不是闹钟，是梦想！

题目描述
给你一个整数n，你需要求出从1×1开始扩展n−1次的格子数量。
每一次扩展，所有与当前格子有相邻（上下左右）的格子都会被扩展进答案里。如下图为n≤4的情形：

输入
输入一个正整数n（1≤n≤106）。
输出
输出从1×1开始扩展n−1次的格子数量。
样例输入 Copy
2
样例输出 Copy
5
提示
对于70%的数据，1≤n≤100。
对于100%的数据，1≤n≤106。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int main()
{
  int n;
    cin>>n;
    if(n==1) printf("1\n");
    else
    {
      ll res=1;
      ll p=4;
      for(int i=0;i<n-1;i++)
      {
            res+=p;
            p+=4;
      }
      cout<<res<<endl;
    }
    return 0;
}