插入排序中元素的移动次数

最新推荐文章于 2024-10-28 10:49:42 发布

原创最新推荐文章于 2024-10-28 10:49:42 发布 · 6k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#插入排序

算法设计专栏收录该内容

49 篇文章

订阅专栏

该博客讨论了插入排序算法中元素的移动次数计算问题。提供了两种解题方法，一是直接通过插入排序过程计算，二是利用分治策略计算逆序对。详细解释了分治策略如何应用于计算，并指出当数组规模减小时，可以快速计算逆序对，与归并排序思路相似，但增加了逆序对计算的步骤。

题目大意：

根据插入排序算法的原理，计算每个元素的移动次数。

例如：数组 $A=[2, 1, 3, 1, 2]$ ，那么按从左往右的顺序依次移动插入元素，他们的移动次数分别如下：

s h i f t (2) = 0, s h i f t (1) = 1, s h i f t (3) = 0, s h i f t (1) = 2, s h i f t (2) = 1

$shift(2)=0, shift(1)=1, shift(3)=0, shift(1)=2, shift(2)=1$
总共需要移动

sum=0+1+0+2+1=4 $sum=0+1+0+2+1=4$ 次。

解题思路：

方法一：按照题目中定义的计算方法（插入排序的原理），逐一计算每个元素的移动次数，即对数组 $A$ 进行一次插入排序。由此可知，时间复杂度为 $O(n^2)$ .
方法二：分治策略，题目中所要求的移动次数等价于数组 $A$ 中的逆序对（比如 $(2,1), (3,2)$ ）的个数。如果将数组一分为二，分别计算左右两边的逆序对数目，再加上右数组相对左数组产生的逆序对数，就是数组的逆序对数。也就说，逆序对存在于左右两数组和左右数组之间。接下来，我们只需要计算左右数组之间的逆序对即可，数组中的逆序对可递归计算。如果左右数组已经排好序，那么计算其之间的逆序对非常快，到这里跟归并排序的思路很像，也是一分为二，进行排序。相对归并排序，我们增加了一项计算逆序对的工作。注意：最小规模下，数组大小为1，逆序对为0。

下面是题解代码

HackerRank:https://www.hackerrank.com/challenges/insertion-sort

#include <cmath>
#include <cstdio>
#include <vector>
#include <iostream>
#include <algorithm>
#include <map>
using namespace std;

long long mergeSort(vector<int>& arr){
    if(arr.size() <= 1){
        return 0;
    }

    int m = arr.size()/2;
    vector<int> v1(arr.begin(), arr.begin()+m);
    vector<int> v2(arr.begin()+m, arr.end());
    long long res(0);
    res += mergeSort(v1);
    res += mergeSort(v2);

    vector<int>::iterator ita, it1, it2;
    ita = arr.begin();
    it1 = v1.begin();
    it2 = v2.begin();
    while(it1!=v1.end() && it2!=v2.end()){
        if(*it1 <= *it2){
            *ita++ = *it1++;
        }
        else{
            *ita++ = *it2++;
            res += (v1.end()-it1);
        }
    }

    while(it1 != v1.end()){
        *ita++ = *it1++;
    }
    while(it2 != v2.end()){
        *ita++ = *it2++;
    }

    return res;
}

int main() {
    /* Enter your code here. Read input from STDIN. Print output to STDOUT */
    int T;
    cin>>T;
    while(T--){
        int n;
        cin>>n;
        vector<int> arr(n);
        for(int i = 0; i < n; i++){
            cin>>arr[i];
        }
        cout<<mergeSort(arr)<<endl;
    }
    return 0;
}