以从小到大排序为例:假设数组中有N个元素,由于每趟冒泡可以将“无序部分”的最大值插入到“有序部分”的最左边,所以需要执行(N - 1)趟冒泡。开始执行第 i 趟冒泡时,“有序部分”包含 i 个元素,“无序部分”包含(N - i)个元素;从“无序部分”的第1个元素开始,与右侧相邻元素比较大小,若左侧元素大于右侧相邻元素,则执行交换;完成(N - i - 1)次比较和交换后,“无序部分”的最大值是其最后一个元素,同时也是“有序部分”的第一个元素。
下面的代码中,若flag为1,则数组按降序排列;否则数组按升序排列。id_sorted中保存了排列顺序。
template<typename T>
void bubble(std::vector<T> &data, int flag, std::vector<int> &id_sorted)
{
id_sorted.clear();
if (data.size()==0) return;
for (int i = 0; i < data.size(); ++i)
id_sorted.push_back(i);
for (int i = 0; i < data.size() - 1; ++i)
{
for (int j = 0; j < data.size() - i - 1; ++j)
{
if (1 == flag)
{
if (data[j] < data[j + 1])
{
double data_tmp = data[j];
data[j] = data[j + 1];
data[j + 1] = data_tmp;
int id_tmp = id_sorted[j];
id_sorted[j] = id_sorted[j + 1];
id_sorted[j + 1] = id_tmp;
}
}
else
{
if (data[j] > data[j + 1])
{
double data_tmp = data[j];
data[j] = data[j + 1];
data[j + 1] = data_tmp;
int id_tmp = id_sorted[j];
id_sorted[j] = id_sorted[j + 1];
id_sorted[j + 1] = id_tmp;
}
}
}
}
}