有一个复数的类模板，有两个私有数据成员，分别是 实部和虚部。有一个成员函数是求该复数的模。 请添加适当的构造函数。
输入格式:
输入仅一行，分别是三个数，以空格间隔。 第一个数是1或2或3（1表示int型，2表示float型，3表示double型），第二个数是该复数的实部，第三个数是该复数的虚部。
输出格式:
输出仅一行，输出该复数的模（in型的复数，输出int型的模; float型的复数，输出float型的模；double型的复数，输出double型的模）。
输入样例:
在这里给出一组输入。例如：
2 2.5 -3.1
输出样例:
在这里给出相应的输出。例如：
3.98246
正确代码：
#include<iostream>
#include<math.h>
using namespace std;
template <class T>
void mod(T* arr){
cin >> arr[0] >> arr[1];
cout << sqrt(arr[0] * arr[0] + arr[1] * arr[1]);
}
int main(){
int a[2];
float b[2];
double c[2];
int flag;
cin >> flag;
if(flag==1){
mod(a);
}else if(flag==2){
mod(b);
}else
mod(c);
return 0;
}