Acwing 基础课—基础算法—归并排序

归并排序模板 稳定O(n*log n) 空间O(n)

参考链接
https://www.acwing.com/solution/content/2099/

#include<stdio.h>
const int N = 100005;
int a[N];
int t[N];
void ms(int l,int r){
    if(l >= r) return ;
    int mid = l + r >> 1;
    ms(l,mid);ms(mid+1,r);
    //merge
    //all is not one,is L
    int i = l;
    int j = mid + 1;
    int k = l;
    while(i <= mid && j <= r){
        if(a[i] < a[j])
            t[k++] = a[i++];
        else
            t[k++] = a[j++];
    }
    while(i <= mid)
        t[k++] = a[i++];
    while(j <= r)
        t[k++] = a[j++];
    for(int i = l;i <= r;i++){
        a[i] = t[i];
    }
}
int main(){
    int n;
    scanf("%d",&n);
    for(int i = 0;i < n;i++) scanf("%d",&a[i]);
    ms(0,n-1);
    for(int i = 0 ;i < n;i++) printf("%d ",a[i]);
    return 0;
}

2 逆序对的数量

题目链接

https://www.acwing.com/problem/content/description/790/

参考链接

https://www.acwing.com/solution/content/5103/

题目大意

给定一个长度为n的整数数列，请你计算数列中的逆序对的数量。
逆序对的定义如下：对于数列的第 i 个和第 j 个元素，如果满足 i < j 且 a[i] > a[j]，则其为一个逆序对；否则不是。

解题思路

在分治后的每一层合并中顺便求出逆序对数量是这个题想法的由来。

while (i <= mid && j <= r)
        if (q[i] <= q[j]) tmp[k ++ ] = q[i ++ ];
        else
        {
            res += mid - i + 1;
            tmp[k ++ ] = q[j ++ ];
        }

• 例如[3,4] [1,2]中，q[0]>q[2],则q[0],q[1]都与q[2]成逆序对,而q[mid]与q[i]有mid-i+1个数字，因此逆序对增加mid-i+1

从而：
（1）只要通过归并排序，将每段构成 【l…mid】【mid+1,…r】的前后各自升序的序列，对于l <= i <= mid,mid+1 <= j <= r,只要发现 a[i] > a[j]，那么对于a[j]，逆序对个数就是 mid - i + 1，【在前面的升序序列中，a[i]大于a[j]，而存在 mid - i 个数大于a[i]，所以共有 mid - i + 1个数大于a[j]，存在mid - i + 1个逆序对】
（2）而每一层的合并操作又构成了前后各自升序的两个序列，又方便了下一次逆序对的计算。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 100005;
int a[N],t[N];
ll ans = 0;

void ms(int l,int r){
    if(l >= r) return ;
    int mid = l + r >> 1;
    ms(l,mid);
    ms(mid+1,r);
    //merge
    int i = l;
    int j = mid + 1;
    int k = l;
    while(i <= mid && j <= r){
        if(a[i] <= a[j]){
            //attention '='
            t[k++] = a[i++];
        }else{
            ans += (mid - i + 1);
            t[k++] = a[j++];
        }
    }
    while(i <= mid) t[k++] = a[i++];
    while(j <= r) t[k++] = a[j++];
    for(int i = l;i <= r;i++){
        a[i] = t[i];
    }
}

int main(){
    int n;
    cin >> n;
    for(int i = 0;i < n;i++) cin >> a[i];
    ms(0,n-1);
    cout << ans << endl;
    return 0;
}