[归并排序]剑指 Offer 51. 数组中的逆序对(hard)

题目:
[归并排序]剑指 Offer 51. 数组中的逆序对(hard)


题解:

思路:归并排序的模板题,在递归处理两个升序子序列时,a[l…r] 中的左序列 a[l…mid] 和右序列 a[mid+1,r] 利用两个指针 i 和 j 在合并时,顺便进行统计逆序对个数。若 a[i]>a[j],由于 a[l…r] 是升序的,那么 a[i…mid]之间的所有元素都会大于 a[j],统计此时的个数为 mid-i+1 即可。


代码如下:

const int N = 5e5+10;
int b[N];
class Solution {
public:
    int reversePairs(vector<int>& a) {
        memset(b,0,sizeof b);
        int n=a.size();
        return merge(a,0,n-1);
    }

    int merge(vector<int>& a,int l,int r)
    {
        if(l>=r)return 0;
        // 1.确定分界点,递归处理左右两段
        int mid=(l+r)>>1;
        int res=merge(a,l,mid)+merge(a,mid+1,r);
        // 2.统计逆序对的个数,顺便合并两个升序子序列
        int i=l,j=mid+1,k=0;
        while(i<=mid||j<=r)
        {
            if(i>mid)b[k++]=a[j++];
            else if(j>r)b[k++]=a[i++];
            else{
                if(a[i]<=a[j])b[k++]=a[i++];
                else{
                    b[k++]=a[j++];
                    // 由于a[i]>a[j],且a[l,mid]是升序的,那么a[i...mid]都会大于a[j]
                    res+=(mid-i+1);
                }
            }
        }
        // 3.还原数组a
        for(i=0;i<k;++i)a[l+i]=b[i];
        return res;
    }
};
上一篇:【工具使用】git使用--将本地代码上传到icode代码库踩坑总结


下一篇:git管理多个分支并进行合并merge等操作记录