关闭

【刷题之路】小范围排序练习

标签: 刷题算法排序
2320人阅读 评论(0) 收藏 举报

已知一个几乎有序的数组,几乎有序是指,如果把数组排好顺序的话,每个元素移动的距离可以不超过k,并且k相对于数组来说比较小。请选择一个合适的排序算法针对这个数据进行排序。给定一个int数组A,同时给定A的大小n和题意中的k,请返回排序后的数组。

因为每个元素移动距离不超过k,可以考虑使用堆排序,比如说数组前K个数中必然有整个数组的最小值,1到k+1个数中必然有第二大的数,以此类推

所以,只要建立一个节点为K的小根堆,每次对K个数进行调整,吐出堆顶,加入第K+1个数,遍历整个数组。排序就完成了

class ScaleSort {
public:
    vector<int> sortElement(vector<int> A, int n, int k) {
        // write code here
        vector<int> B(k);
        int i,j;
        for(i=0;i<k;i++)  //复制A给B
            B[i]=A[i];
             
        minheap(B,k); //建立小根堆
          
        for(i=0;i<n-k;i++) //从第一个到n-k,一共有n-k个K节点的小根堆
            {
            A[i]=B[0];  //堆顶即为有序部分,存入A中
            B[0]=A[i+k]; //加入k+1
            shiftdown(B,0,k); //重新调整小根堆
        }
          
        for(j=n-k;j<n;j++) //对于剩下的K-1个数,已经没有可以加入的数字了,不断减小小根堆的节点数直到为0
            {
             A[j] =B[0];
            if(j==n-1)
                break;
            B[0]=B[--k];
            shiftdown(B,0,k);
        }
              
         return A;
    }
    void minheap(vector<int> &a,int n){
        int i=n/2-1;
        for(i;i>=0;i--){
            shiftdown(a,i,n);
        }
    }
    void shiftdown(vector<int> &a,int i,int n){
        int j=0;
        int flag=1;
        while(i*2+1<n && flag){  
            if( i*2+2< n && a[i*2+2] < a[i*2+1] ) //与之前的大根堆略有不同,小根堆是寻找两个子节点中较小的
                j=i*2+2;
            else
                j=i*2+1;
            if(a[i] > a[j]){  //如果新加入的节点比较小的节点大,则调整两个数字,将较小的数字往上移动
                int temp=a[i];
                a[i]=a[j];
                a[j]=temp;
                i=j;
            }
            else
                flag=0;
        }
    }
};

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:11739次
    • 积分:811
    • 等级:
    • 排名:千里之外
    • 原创:72篇
    • 转载:2篇
    • 译文:0篇
    • 评论:1条