几乎有序是指,如果把数组排好顺序的话,每个元素移动的距离可以不超过k,并且k相对于数组来说比较小。请选择一个合适的排序算法针对这个数据进行排序。 给定一个int数组A,同时给定k
1、使用堆排序,每次从后开始的k个数,建立堆,然后找到最大的放后边,然后不断的建堆,最后对第1个数进行处理即可(ps:因为第一个数没参与堆排序),每次建堆为lok(k),所以总共的时间复杂度为O(nlog(k))
2、使用直接插入排序,因为直接插入排序是和之间的元素是相关的,每个元素最多移动k位,所以时间复杂度为O(nk)
//n为1、2、3....即以1开始的实际位置,k代表移动位数
void HeapSort(vector<int> &vec, int n,int k)
{
int ini = n;
int siz = n+k-1;
int num = n+k/2-1;
while (n <= num)
{
//考虑到最后靠右的时候的非叶子节点只有一个左节点的情况
if (n == num)
{
if (n+k/2 == siz)
{
if (vec[n - 1] < vec[siz-1])
swap(vec[n - 1], vec[siz-1]);
}
else
{
//x为以0开始的数组位置
int x = vec[n - 1 +k/2] > vec[n + k / 2] ? n - 1 + k / 2 : n + k / 2;
if (vec[n-1] < vec[x])
swap(vec[n-1], vec[x]);
}
break;
}
else
{
//x为以0开始的数组位置
int x = vec[2 * (n-ini+1)+ini-2] > vec[2 * (n - ini + 1) + ini -1] ? 2 * (n - ini + 1) + ini - 2 : 2 * (n - ini + 1) + ini - 1;
if (vec[n - 1] < vec[x])
{
swap(vec[n - 1], vec[x]);
n = x + 1;
}
else
break;
}
}
}
void SortElem(vector<int> &vec,int k)
{
vector<int> ans;
int n = vec.size();
for (int i = n-k+k/2; i > n - k; i--)
HeapSort(vec, i,k);
for (int i = n - k; i > 0; i--)
{
//堆排好后进行位置交换
int temp = vec[i];
for (int j = 0; j < k-1; j++)
{
vec[i + j] = vec[i + j + 1];
}
vec[i + k - 1] = temp;
//重新堆排序i、i+1、、、i+k-1
HeapSort(vec, i, k);
}
//对前k个数进行处理
int temp = vec[0];
for (int i = 1; i < k; i++)
{
if (temp < vec[i])
break;
else
swap(vec[i - 1], vec[i]);
}
//for (auto x : vec) cout << x << " ";
}