樓蘭圖騰 Page205 樹狀數組求逆序對

樓蘭圖騰 Page205 樹狀數組求逆序對

1.樹狀數組寫起來感覺比歸併好理解多了
2.求^和v的形狀,求出任意一個後,可以採用將數組“倒過來”的方法,即將數組的大小關係重置一下
3.不僅是逆序對,正序對也很好求,改變遍歷方向即可

代碼:

/**
 *  Author1: low-equipped w_udixixi
 *  Author2: Sher丶lock
 *  Date :2020-07-03
**
Zz:overint or overarray
Xx:overwatch to 5e8
Ss:unordered map
Ww:memset or f_o_r
Ii:vector or neighbours
**
**/
#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<queue>
#include<map>
#define ll long long
#define ull unsigned long long
#define pb push_back
#define rep(x,a,b) for (int x=a;x<=b;x++)
#define repp(x,a,b) for (int x=a;x<b;x++)
#define W(x) printf("%d\n",x)
#define WW(x) printf("%lld\n",x)
#define pi 3.14159265358979323846
#define mem(a,x) memset(a,x,sizeof a)
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
using namespace std;
const int maxn=2e6+7;
const int INF=1e9;
const ll INFF=1e18;
int a[maxn],c[maxn],Lsmall[maxn],Rsmall[maxn],n;
void add(int x,int y){for (;x<=n;x+=x&-x)c[x]+=y;}
int ask(int x){int ans=0;for (;x;x-=x&-x)ans+=c[x];return ans;}
ll solve()
{
    ll cnt=0;mem(c,0);
    for (int i=n;i;i--)
    {
        Rsmall[i]=ask(a[i]-1);
        add(a[i],1);
    }
    mem(c,0);
    rep(i,1,n)
    {
        Lsmall[i]=ask(a[i]-1);
        add(a[i],1);
        cnt+=1ll*Lsmall[i]*Rsmall[i];
    }
    return cnt;
}
int main()
{
    ll ans1=0,ans2=0;
    scanf("%d",&n);
    rep(i,1,n)scanf("%d",&a[i]);
    ans1=solve();
    rep(i,1,n)a[i]=n+1-a[i];
    ans2=solve();
    printf("%lld %lld\n",ans2,ans1);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章