您的位置:首页 > 其它

用堆排序实现大致已经排好序的序列

2017-02-13 00:33 155 查看
已知一个几乎有序的数组,几乎有序是指,如果把数组排好顺序的话,每个元素移动的距离可以不超过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]

可以利用堆排序,不超过k,每组选择k+1个数,从0开始,每组内进行建堆,建完最小堆之后最前面的元素就是最小值。之后下标加1,继续进行堆排序,时间复杂度可以达到(log(n+k))空间复杂度为O(1)因为使用循环实现最小堆,没用调用栈。不稳定排序。

代码如下:

class ScaleSort {
public:
vector<int> sortElement(vector<int> A, int n, int k) {
if(n<2)
return A;
int i=0;
for(;i+k<n;++i)
{
buildheap(A,i,k+1);
}

while(n-i>1)
{
buildheap(A,i,n-i);
i++;
}
return A;
// write code here
}

void buildheap(vector<int> &A,int begin,int k)
{
if(k<2)
return;
if(A[(k-2)/2+begin]>A[k-1+begin])
swapnum(A[(k-2)/2+begin],A[k-1+begin]);

if(k%2==1){
if(A[(k-2)/2+begin]>A[k-2+begin])
swapnum(A[(k-2)/2+begin],A[k-2+begin]);
}
for(int i=(k-4)/2;i>=0;--i)
{
if(A[i+begin]>A[2*i+1+begin])
swapnum(A[i+begin],A[2*i+1+begin]);
if(A[i+begin]>A[2*i+2+begin])
swapnum(A[i+begin],A[2*i+2+begin]);
}
}
void swapnum(int &a,int &b)
{
int temp;
temp=a;
a=b;
b=temp;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: