1366: 【入门】平面分割问题

#include<bits/stdc++.h>
using namespace std;
 
int cal(int x){
    int s=2;
    for (int i = 1; i <= x; i++) {
        s = s + (i-1) * 2;
 
    }
    return s;
 
}
int main(){
    int m;
    cin>>m;
    cout<<cal(m);
 
}