【C++ | PTA】2017final函数模板

题目要求

数据的间距问题（函数模板） 类point有三个数据成员：x、y和z, 分别代表x坐标、y坐标和z坐标，并有若干构造函数和一个重载-（减号，计算两点距离）的成员函数。 要求设计一个函数模板，
template < class T> double dist(T a, T b)
对int，float，point或者其他类型的数据，返回间距。

输入格式:
每一行为一个操作，每行的第一个数字为元素类型，1为整型元素，2为浮点型元素，3为point类型，若为整型元素，接着输入两个整型数据，若为浮点型元素，接着输入两个浮点型数据，若为point型元素，输入两个point型数据（x1 y1 z1 x2 y2 z2），输入0时标志输入结束。

输出格式:
对每个输入，每行输出一个间距值。

输入样例:
1 2 5
3 2 4 7 5 9 7
2 2.2 9.9
0

输出样例:
3
5.83095
7.7

代码

#include<iostream>
#include<cmath>
using namespace std;
// 定义 Point 类表示三维坐标点
class Point{
private:
    double x;
    double y;
    double z;
public:
    // Point 类构造函数，用于初始化 x、y、z 三个数据成员
    Point(double a,double b,double c):x(a),y(b),z(c){};   
    // 友元函数，实现两个 Point 对象之间的距离计算
    friend double operator-(Point,Point);            
};
// 函数模板，用于计算任意类型的数据之间的距离
template <class T>
double dist(T a, T b){                 
    return abs(a-b);
}
// 重载 - 运算符，计算两个 Point 对象之间的距离
double operator-(Point P1,Point P2){ 
    return sqrt(pow(P1.x-P2.x,2)+pow(P1.y-P2.y,2)+pow(P1.z-P2.z,2));   
}
// 主函数入口
int main(){
    int i, j;
    float p, q;
    double x1, y1, z1, x2, y2, z2;
    int flag;
    while(1){
        cin >> flag;
        if(flag == 0){
            break;            
        }
        if(flag == 1){
            cin >> i >> j;
            // 调用 dist 模板函数计算 i 和 j 之间的间距，并输出
            cout << dist(i, j) << endl;
        } else if(flag == 2){
            cin >> p >> q;
            // 调用 dist 模板函数计算 p 和 q 之间的间距，并输出
            cout << dist(p, q) << endl;
        } else{
            cin >> x1 >> y1 >> z1 >> x2 >> y2 >> z2;
            Point P1(x1, y1, z1), P2(x2, y2, z2);
            // 调用 operator- 函数计算 P1 和 P2 之间的距离，并输出
            cout << dist(P1, P2) << endl;
        }
    }
    return 0;
}