数据结构与算法——插入排序——使用C++实现函数模板

1. 插入排序原理

插入排序是最基本的排序算法，之前写过一篇文章实现插入排序，不过是用的C语言+函数指针，使得用户可以自定义升序排序还是降序排序。

现在考虑使用C++，将该算法做成一个函数模板。

2. 将排序算法做成函数模板

几个关键字：【函数模板】，【函数对象】，【匿名函数】，【友元函数】。

对于基本数据类型，构建函数模板比较简单，排序的比较也可以直接用STL里面的less和great函数对象。对于自定义类型的数据，需要实现自定义类型对象的比较，可以1）重载比较运算符，less重载"<"运算符，great重载">"运算符；2）实现一个比较用的函数对象，这个类要操作自定义类型数据的成员，因此可以是友元类；3）跟2）类似，不过是用匿名函数(lambda表达式).

以下是完整代码，具体原理见注释。

/**
 * @file InsertSortTemplate.cpp
 * @brief 插入排序基本思想与整扑克牌类似，将未排序元素中最前面的那个插入到已排序序列的合适位置，然后需要将已排序序列中该位置后面的元素向后挪一个元素位置
 1. 基于基本思想的简单方法：
   （1）在当前有序区R[1...i-1]中查找R[i]的正确插入位置k(1<=k<=i-1)；
   （2）将R[k...i-1]中的记录均后移一个位置，腾出k位置上的空间插入R[i]。
2. 改进方法：查找比较和记录移动交替进行：
   将待插入记录R[i]的关键字从右向左依次与有序区中记录R[j](j=i-1,i-2,...,1)的关键字进行比较：
   （1）如果R[j]的关键字大于R[i]的关键字，则将R[j]后移一个位置；
   （2）如果R[j]的关键字小于或等于R[i]的关键字，则查找过程结束，j+1即为R[i]的插入位置。
	实现了将插入排序函数封装为函数模板，可以对任意类型数据进行排序
 * @author 好好学习
 * @version v1
 * @date 2020-07-31
 */

#include <iostream>
#include <vector>
#include <ctime>
using namespace std;

template<typename T, class compare = std::less<T>>
void insertSort(vector<T>& nums, compare cmp = std::less<T>());//实现插入排序功能的函数

template<typename T>
void printVector(vector<T>& nums);//辅助函数：打印容器中元素

void generateIntVector(vector<int>& nums, int n);//辅助函数：生成n个随机数填入int类型的vector

//该类用来测试insertSort函数对自定义数据类型排序的正确性
//对自定义数据类型，需要1）重载小于或者大于运算符；2）自定义用于比较的函数对象
class Base {
	int x;
public:
	Base() :x(-1) {}
	explicit Base(int m_x) :x(m_x) {}
	friend bool operator< (const Base& b1, const Base& b2);
	friend class cmp1;
	friend ostream& operator<<(ostream& os, const Base& b);
	int getX() { return x; }
};
//重载小于"<"运算符
bool operator<(const Base & b1, const Base & b2){
	return b1.x < b2.x;
}
//重载<<，用于打印输出
ostream& operator<<(ostream& os, const Base& b) {
	os << b.x;
	return os;
}
//定义函数对象，用于自定义类型的比较
struct cmp1 {
	bool operator()(Base& b1, Base& b2) {
		return b1.x < b2.x;
	}
};

int main()
{
	srand((int)time(0));
	//测试基本数据类型的排序
	vector<int> nums1;
	generateIntVector(nums1, 10);
	cout << "Initial value of nums1: ";
	printVector<int>(nums1);

	insertSort<int>(nums1,greater<int>());

	cout << "Sorted value of nums1: ";
	printVector<int>(nums1);

	//以下测试自定义数据类型的排序
	vector<Base> baseArr;
	for (int i = 0; i < 15; i++) {
		int tmpVal = rand() % 100;
		Base tmpBase(tmpVal);
		baseArr.push_back(tmpBase);
	}
	cout<<"Initial value of baseArr: ";
	printVector<Base>(baseArr);
	//insertSort<Base>(baseArr);//重载<运算符之后，使用默认的less
	//insertSort<Base>(baseArr, cmp1());//使用自定义的函数对象进行比较
	insertSort<Base>(baseArr, [](Base b1, Base b2)mutable -> bool {return b1.getX() < b2.getX(); });//使用匿名函数进行比较
	cout << "Sorted value of baseArr: ";
	printVector<Base>(baseArr);
	return 0;
}
/**
 * @brief 向标准输出流输出vecotr中的元素，元素之间以空格分割
 * @param[in]  nums  要打印的vector
 * @return
 */
template<typename T>
void printVector(vector<T>& nums) {
	if (nums.empty()) {
		cout << "The vector is empty!" << endl;
		return;
	}
	cout << nums[0];
	for (int i = 1; i < nums.size(); i++)cout << " " << nums[i];
	cout << endl;
}
/**
 * @brief 生成n个int型随机数，填充到vector容器中
 * @param[in]   n  要填充的元素的个数
 * @param[out]  nums   要往里填充元素的容器
 * @return
 */
void generateIntVector(vector<int>& nums, int n) {
	int tmp = 0;
	for (int i = 0; i < n; i++) {
		tmp = rand() % 100;
		nums.push_back(tmp);
	}
}

/**
 * @brief 将vector容器中的元素进行排序
 * @param[in]  nums  要排序的vector
 * @return
 */
template<typename T, class compare>
void insertSort(vector<T>& nums, compare cmp) {
	int n = nums.size();
	int i = 1, j = 1;
	T tmp;
	for (; i < n; i++) {
		tmp = nums[i];
		for (j = i - 1; j >= 0 && cmp(tmp, nums[j]); j--) {//如果nums[j]大于nums[i]
			nums[j + 1] = nums[j];	//将nums[j]后移
		}
		//内层循环结束之后，nums[j]<tmp，此时nums[j+1]是tmp的位置
		nums[j + 1] = tmp;
	}
}