树状数组求逆序对

题目：http://acm.hdu.edu.cn/showproblem.php?pid=5792

分析：

计算像a，b这样上升的有la对，像c，d这样下降的有lb对，ans=la*lb。这样是有重复的，重复的就是a与c重合，a与d重合，b与c重合，b与d重合这四种情况。那么减去这四种情况就ok了。

可以用树状数组预处理出每一位i的左边比a[i]大的有多少La[]，少的有多少Li[]，右边比a[i]大的有多少Ra[]，少的有多少Li[]。最后ans减去Ri[i]*Ra[i];    Li[i]*La[i];   Li[i]*Ri[i];   La[i]*Ra[i]  就行了。

Ac code:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=5e4+5;
int b[maxn];
ll Ll[maxn],Lr[maxn],Rl[maxn],Rr[maxn];
int k;
struct Node{
   int val;
   int id;
   bool operator<(const Node& a)const
   {
       return val<a.val;
   }
};
Node a[maxn];
int lowbit(int x)
{
    return x&(-x);
}
void add(int pos,int val,ll c[])
{
    while(pos<maxn)//注意是<maxn,而不是n
    {
        c[pos]+=val;
        pos+=lowbit(pos);
    }
}
ll query(int pos,ll c[])
{
    ll sum=0;
    while(pos>0)
    {
        sum+=c[pos];
        pos-=lowbit(pos);
    }
    return sum;
}
ll c1[maxn],c2[maxn];
int cnt[maxn];///用于计算重复的数的个数
int main()
{
    int n;
    while(~scanf("%d",&n)){
        memset(c1,0,sizeof c1);
        memset(c2,0,sizeof c2);
        memset(cnt,0,sizeof cnt);
        for(int i=1;i<=n;i++) scanf("%d",&a[i].val),a[i].id=i;
        sort(a+1,a+1+n);
        k=1;
        for(int i=1;i<=n;i++)
        {
            if(i!=1&&a[i].val!=a[i-1].val) ++k;
            b[a[i].id]=k;///离散化
        }
        ll sum1=0,sum2=0;
        for(int i=1;i<=n;i++)
        {
            add(b[i],1,c1);
            ll num=query(b[i],c1);
            cnt[b[i]]++;
            Lr[i]=i-num;///左边比b[i]大的数的个数=插入的数的总个数-小于等于b[i]的数的个数
            Ll[i]=num-cnt[b[i]];///左边比b[i]小的数的个数,注意要减去cnt[b[i]]而不是1,因为前面也可能已经出现了b[i]
            sum1+=Lr[i];///逆序对数
        }
        memset(cnt,0,sizeof cnt);
        for(int i=n;i>=1;i--)
        {
            add(b[i],1,c2);
            ll num=query(b[i],c2);
            cnt[b[i]]++;
            Rr[i]=(n-i+1)-num;
            Rl[i]=num-cnt[b[i]];
            sum2+=Rr[i];///正序对数
        }
        ll ans=sum1*sum2;
        for(int i=1;i<=n;i++)
        {
            ans=ans-Rl[i]*Rr[i];
            ans=ans-Ll[i]*Lr[i];
            ans=ans-Ll[i]*Rl[i];
            ans=ans-Lr[i]*Rr[i];
        }
        printf("%lld\n",ans);
    }
    return 0;
}

最后贴个树状数组求逆序对的板子：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
using namespace std;
 
int n,tree[100010];
void add(int k,int num)
{
    while(k<=n)
    {
        tree[k]+=num;
        k+=k&-k;
    }
}
 
int read(int k)
{
    int sum=0;
    while(k)
    {
        sum+=tree[k];
        k-=k&-k;
    }
    return sum;
}
struct node
{
    int val,pos;
}a[100010];
bool cmp(node a,node b)
{
    return a.val < b.val;
}
int main(void)
{
    int i,j;
    int b[100010];
    while(scanf("%d",&n)==1)
    {
        memset(tree,0,sizeof(tree));
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i].val);
            a[i].pos = i;
        }
        sort(a+1,a+1+n,cmp);
        int cnt = 1;
        for(i=1;i<=n;i++)
        {
            if(i != 1 && a[i].val != a[i-1].val)
                cnt++;
            b[a[i].pos] = cnt;
        }
        LL sum = 0;
        for(i=1;i<=n;i++)
        {
            add(b[i],1);
            sum += (i - read(b[i]));
        }
        printf("%lld\n",sum);
    }
 
    return 0;
}

 

外加归并排序求逆序对的板子：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
ll ans;
void Merge(int a[],int l,int mid,int r)
{
    int b[100005];
    for(int i=l; i<=r; i++)
        b[i]=a[i];
    int i,j,k;
    for(i=l,j=mid+1,k=l; i<=mid&&j<=r; ++k)
        if(b[i]<=b[j])
        {
            a[k]=b[i];
            ++i;
        }
        else
        {
            a[k]=b[j];
            ++j;
            ans+=mid-i+1;///若有序表b中,b[i]>b[j],那么[i,mid]这个区间都比b[j]大，逆序对数mid-i+1
        }
    while(i<=mid) a[k++]=b[i++];
    while(j<=r) a[k++]=b[j++];
}
void divide(int a[],int l,int r)
{
    if(l>=r)
        return;
    int mid=(l+r)>>1;
    divide(a,l,mid);
    divide(a,mid+1,r);
    Merge(a,l,mid,r);
}
int main()
{
    int a[maxn];
    int x,n;
    while(~scanf("%d",&n))
    {
        ans=0;
        for(int i=1; i<=n; i++)
            scanf("%d",&a[i]);
        divide(a,1,n);
        //for(int i=1; i<=n; i++) printf("%d%c",a[i],i==n?'\n':' ');
        printf("%lld\n",ans);
    }
    return 0;
}