PTA 7-1 数据的间距问题 (20分) c++

复数类Complex有两个数据成员：a和b, 分别代表复数的实部和虚部，并有若干构造函数和一个重载-（减号，用于计算两个复数的距离）的成员函数。 要求设计一个函数模板

template < class T >
double dist(T a, T b) 

对int，float，Complex或者其他类型的数据，返回两个数据的间距。
以上类名和函数模板的形式，均须按照题目要求，不得修改
输入格式:
每一行为一个操作，每行的第一个数字为元素类型，1为整型元素，2为浮点型元素，3为Complex类型，若为整型元素，接着输入两个整型数据，若为浮点型元素，接着输入两个浮点型数据，若为Complex型元素，输入两个Complex型数据（a1 b1 a2 b2），输入0时标志输入结束。
输出格式:
对每个输入，每行输出一个间距值。
输入样例:

1 2 5
3 2 4 5 9
2 2.2 9.9
0

输出样例:

3
5.83095
7.7

AC代码

#include<cmath>
#include<cstdio>
#include<algorithm>
#include<string>
#include<vector>
#include<iomanip>
#include<iostream>
using namespace std;
typedef long long ll;
class Complex{

public :
    Complex(double a1=0,double b1=0){
        a=a1;
        b=b1;
    }
    friend double operator - (Complex a, Complex b);
private:
    double a,b;

};
double operator - (Complex a, Complex b){

    return sqrt(pow(a.a - b.a, 2.0) + pow(a.b - b.b, 2.0));
}
template <class T>
double dist(T a, T b){
    return abs(a-b);
}

int main() {

     int n;
     cin>>n;
     while(n){
        if(n==1){
            int a,b;
            cin>>a>>b;
            cout<<dist(a,b)<<endl;
        }
        else if(n==2){
            float a,b;
            cin>>a>>b;
            cout<<dist(a,b)<<endl;
        }
        else if(n==3){
            double a1,b1,a2,b2;
            cin>>a1>>b1>>a2>>b2;
            Complex a(a1,b1),b(a2,b2);
            cout<<dist(a,b)<<endl;
        }
        cin>>n;
     }

      return 0;
}