giúp mình với, viết bằng python và c++

Python

n = int(input())
print(n*n)

C++

#include <iostream>
using namespace std;
int main() {
long long n;
cin >> n;
cout << n*n;
return 0;
}

def summ(n) : 
    S = 0
    for i in range(1,2*n,2) : 
        S+=i 
    return S 
n = int(input())
print(summ(n))

_________________________________________

#include<bits/stdc++.h>

using namespace std;

int main(){

     long long n,S;

     S = 0;

     for(int i = 1 ; i < 2*n ; i+=2){S += i;}

      cout << S;

}