问题引入
如果要写一个函数实现两个数的互换可以这样写:
void mySwap(int &a, int &b){
int temp;
temp = a;
a = b;
b = temp;
}
但是我们要交换的数字不只是int
类型,也有可能是double
或者float
等类型,那么就需要定义三个这样的函数:
void mySwapInt(int &a, int &b){
int temp;
temp = a;
a = b;
b = temp;
}
void mySwapDouble(double &a, double &b){
double temp;
temp = a;
a = b;
b = temp;
}
void mySwapFloat(float &a, float &b){
float temp;
temp = a;
a = b;
b = temp;
}
由于C++支持函数重载,可以支持函数名称相同但是参数不同(C语言不支持),所以可以这样写:
#include <iostream>
void mySwap(int& a, int& b) {
int temp;
temp = a;
a = b;
b = temp;
}
void mySwap(double& a, double& b) {
double temp;
temp = a;
a = b;
b = temp;
}
void mySwap(float& a, float& b) {
float temp;
temp = a;
a = b;
b = temp;
}
void main() {
int a = 10;
int b = 20;
mySwap(a, b);
std::cout << "a = " << a << std::endl;
std::cout << "b = " << b << std::endl;
}
但是这样写起来仍然要将函数定义三次,还是很麻烦,C++提供了函数模板的功能来解决这个问题。
函数模板
下面我们定义一个简单的函数模板:
这里的T
代表一种抽象的数据类型,可以用其他字母,但是习惯上使用字母T
。
template <class T>
void mySwap(T &a, T &b){
T temp;
temp = a;
a = b;
b = temp;
}
完整代码:
#include <iostream>
template <class T>
void mySwap(T& a, T& b) {
T temp;
temp = a;
a = b;
b = temp;
}
void main() {
int a = 10;
int b = 20;
mySwap(a, b);
std::cout << "a = " << a << std::endl;
std::cout << "b = " << b << std::endl;
}