2017final函數模板

數據的間距問題（函數模板） 類point有三個數據成員：x、y和z, 分別代表x坐標、y坐標和z坐標，並有若干構造函數和一個重載-（減號，計算兩點距離）的成員函數。 要求設計一個函數模板，

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

對int，float，point或者其他類型的數據，返回間距。

#include<iostream>
#include<cmath>
using namespace std;

class Point
{
private:
    double x, y, z;
public:
    Point()
    {
        x = 0, y = 0, z = 0;
    }
    Point(int xx, int yy, int zz): x(xx), y(yy), z(zz) {}
    void set(int xx, int yy, int zz)
    {
        x = xx, y = yy, z = zz;
    }
    friend istream& operator >> (istream &input, Point &P);
    friend double operator - (Point P1, Point P2);
};

istream& operator >> (istream &input, Point &P)
{
    input >> P.x >> P.y >> P.z;
    return input;
}

double operator - (Point P1, Point P2)
{
    return sqrt(pow(P1.x - P2.x, 2.0) + pow(P1.y - P2.y, 2.0) + pow(P1.z - P2.z, 2.0));
}

template<class T>
double dist(T a, T b)
{
    return abs(a - b);
}

int main()
{
    int flag;

    while(cin >> flag, flag != 0)
    {
        if(flag == 1)
        {
            int a, b;
            cin >> a >> b;
            cout << dist(a, b) << endl;
        }
        else if(flag == 2)
        {
            float a, b;
            cin >> a >> b;
            cout << dist(a, b) << endl;
        }
        else if(flag == 3)
        {
            Point a, b;
            cin >> a >> b;
            cout << dist(a, b) << endl;
        }
    }

    return 0;
}