编写一个程序，既能计算长方形的面积，又能计算正方形的面积

#include<iostream>
using namespace std;
class sq
{
private: int x,y;
public:
int mysqrt(int a);
int mysqrt(int a,int b);
};
int sq::mysqrt(int a)

{ x=a;
return(x*x);
}
int sq::mysqrt(int a,int b)
{ x=a;
y=b;
return(x*y);
}
int main()
{
int n,a,b;
sq mm;
cout<<"你想计算长方形还是正方形？ 正方形选择1，长方形选择2"<<endl;
cin>>n;
if(n==1)
{
cout<<"输入正方形的边长：";
cin>>a;
cout<<"正方形的面积是"<<mm.mysqrt(a)<<endl;
}
else if(n==2)
{
cout<<"输入长方形的边长：";
cin>>a>>b;
cout<<"长方形的面积是"<<mm.mysqrt(a,b)<<endl;
}
else
cout<<"选择错误："<<endl;
return 0;
}