打卡 函数模板实现复数的重载运算

复数类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

 

思路：主要是要利用模板实现代码复用，然后通过函数重载实现对不同数据类型的相加减。

　　　同时在主函数部分去判断输入的数据类型，通过switch选择响应的输出方式

代码实现：

#include<iostream>

#include<cmath>

#include<iomanip>

using namespace std;

template<class T>

double dist(T a, T b){

    return abs(b-a);

}

class Complex{

    private:

     double real,image;

    public:

     Complex(double a=0,double b=0):real(a),image(b){};

     double operator-(const Complex &c){

         double temp;

         temp=sqrt((real-c.real)*(real-c.real)+(image-c.image)*(image-c.image));

         return temp;

     }

};

int main(){

    int type;

    cin>>type;

    while(type){

        switch (type)

        {

        case 1:{

            int a,b;

            cin>>a>>b;

            cout<<dist(a,b)<<endl;

            break; 

        }

        case 2:{

            float f1,f2;

            cin>>f1>>f2;

            cout<<dist(f1,f2)<<endl;

            break;

        }

        case 3:{

            double real1,image1,real2,image2;

            cin>>real1>>image1>>real2>>image2;

            Complex c1(real1,image1),c2(real2,image2);

            cout<<dist(c1,c2)<<endl;

            break;

        }

        default:

            break;

        }

        cin>>type;

    }

}