题目:
已知一个几乎有序的数组,几乎有序是指,如果把数组排好顺序的话,每个元素移动的距离可以不超过k,并且k相对于数组来说比较小。请选择一个合适的排序算法针对这个数据进行排序。
给定一个int数组A,同时给定A的大小n和题意中的k,请返回排序后的数组。
测试样例:
[2,1,4,3,6,5,8,7,10,9],10,2
返回:[1,2,3,4,5,6,7,8,9,10]
基本思想:
1. 依题目可知如果将数组分成n/k个序列,这个n/k个序列看成一个整体话,这个N/k个整体前面的数的最大值应该小于后面那个整体的最小值,即N个整体应该是有序的。
2. 故可以先取前面k个数建立一个最小堆。对后面n-k个数依次加入最小堆,取最小堆的堆顶值之后,维护堆。
3. 这样可以得到数组中排序的前面n-k个数。之后把最后最小堆的k个数加入数组,则可以得到排序的整个数组。
代码如下:
class ScaleSort {
public:
vector<int> sortElement(vector<int> A, int n, int k) {
// write code here
vector<int> B(k);
for(int i=0;i<k;i++)
B[i]=A[i];
for(int i=k/2-1;i>=0;--i)//维护一个k个数的最小堆
MinHeapAdjust(B,i,k);
for(int i=k;i<n;++i)//取出数组前面最小的n-k个数
{
A[i-k]=B[0];//取出最小值,依次赋值给A数组
B[0]=A[i];
MinHeapAdjust(B,0,k);
}
for(int i=n-k; i<n; i++)//后面的k个数即是数组的最大的k个数
{
A[i] = B[0];
int temp = B[k-1];
B[k-1] = B[0];
B[0] = temp;
MinHeapAdjust(B,0,--k);
}
return A;
}
void MinHeapAdjust(vector<int> &A,int parent,int nLength)
{
int temp;
int child;
for(;2*parent+1<nLength;parent=child)
{
child=2*parent+1;
if(child+1<nLength&&A[child+1]<A[child]) ++child;
if(A[parent]>A[child])
{
int temp=A[parent];
A[parent]=A[child];
A[child]=temp;
}
else break;
}
}
};