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

已知一个几乎有序的数组，几乎有序是指，如果把数组排好顺序的话，每个元素移动的距离可以不超过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;
        }
    }
};