歸併排序求逆序對(不會樹狀數組。。。)

Powered by:AB_IN 局外人

P1908 逆序對

對於給定的一段正整數序列,逆序對就是序列中 ai>aji<j的有序對。
其實就是冒泡排序時,將數組升序排列時交換的次數。但n≤5×10^5 ,O(n^2)肯定不行 。看題解 可以用樹狀數組做,但菜雞還沒學。
so,歸併排序。(跑了接近一秒,不是最優解哈)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#pragma GCC optimize(2)
#pragma GCC optimize(3)
const ll maxn=1e6+10;
ll n,cnt,a[maxn],b[maxn];
char buf[1 << 21], *p1=buf, *p2=buf;
inline ll getc(){
    return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 21, stdin), p1 == p2) ? EOF : *p1++;
}
inline ll read() {
    ll ret = 0,f = 0;char ch = getc();
    while (!isdigit (ch)) {
        if (ch == '-') f = 1;
        ch = getc();
    }
    while (isdigit (ch)) {
        ret = ret * 10 + ch - 48;
        ch = getc();
    }
    return f ? -ret : ret;
}
inline void write(ll x) { if (!x) { putchar('0'); return; } char F[200]; ll tmp = x > 0 ? x : -x; if (x < 0)putchar('-'); int cnt = 0;    while (tmp > 0) { F[cnt++] = tmp % 10 + '0';     tmp /= 10; }    while (cnt > 0)putchar(F[--cnt]); }
void _merge(ll l,ll mid,ll r)
{
    ll p1=l,p2=mid+1;
    for(ll i=l;i<=r;i++){
        if((p1<=mid) && ((p2>r) || a[p1] <= a[p2])){
            b[i]=a[p1];
            p1++;
        }
        else{
            b[i]=a[p2];
            p2++;
            cnt+=mid-p1+1;//最關鍵的一句
        }
    }
    for(ll i=l;i<=r;i++) a[i]=b[i];
}
void erfen(ll l,ll r)
{
    ll mid=(l+r)/2;
    if(l<r){
        erfen(l,mid);
        erfen(mid+1,r);
    }
    _merge(l,mid,r);
}
int main()
{
    n=read();
    for(ll i=1;i<=n;i++)
        a[i]=read();
    erfen(1,n);
    write(cnt);putchar(10);
    return 0;
}

歸併排序的原理要是知道了,那是什麼時候像冒泡排序一樣消除的逆序對呢?
對,就是這個時候。

else{
            b[i]=a[p2];
            p2++;
            cnt+=mid-p1+1;//最關鍵的一句
        }

什麼意思呢,看下面的圖,就是讓a數組右區間在p2的值提到了前面(左區間),而紅色塗滿的區間都與p2指針指的數構成逆序對,故求區間長度就可以了。
在這裏插入圖片描述
在這裏插入圖片描述
完結。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章