2018final复数求模的类模板

有一个复数的类模板，有两个私有数据成员，分别是 实部和虚部。有一个成员函数是求该复数的模。 请添加适当的构造函数。

输入格式:

输入仅一行，分别是三个数，以空格间隔。 第一个数是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;
}