LA 4329 BIT 分治

#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#define lowbit(x) x&-x

using namespace std;

int c[100010],a[20010],n,T,l[20010];

int sum(int x)
{
    int ret=0;
    while(x>0)
    {
        ret+=c[x];
        x-=lowbit(x);
    }
    return ret;
}

void add(int x,int d)
{
    while(x<=100000)
    {
        c[x]+=d;
        x+=lowbit(x);
    }
}

int main()
{
    //freopen("/home/user/in","r",stdin);
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        memset(c,0,sizeof(c));
        long long s=0;
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
            add(a[i],1);
        //    printf("%d
",c[i]);
            l[i]=sum(a[i]-1);//l[i]表示a[i]前面比a[i]小的数的个数
        }
        //for(int i=0;i<n;i++)
        //    printf("a[%d]=%d %d %d
",i,a[i],l[i],sum(a[i]));
        for(int i=0;i<n;i++)
        {
            int al=sum(a[i]-1);//al表示整个a数组中比a[i]小德数的个数
            s+=l[i]*(n-i-1-al+l[i])+(i-l[i])*(al-l[i]);
        }
        printf("%lld
",s);
    }
    return 0;
}

用BIT做的

#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <set>

using namespace std;

struct Node
{
    int i,x;
    bool operator<(Node &a)const
    {
        return x<a.x;
    }
};

Node a[20010],t[20010];
int c[20010],n,T;

void merge_sort(int l,int r)
{ 
    if(r-l>1)
     { 
        int m=l+(r-l)/2;
        int p=l,q=m,i=l;
        merge_sort(l,m);
        merge_sort(m,r);
        while(p<m||q<r)
        {
            if(q>=r||(p<m&&a[p]<a[q])) memcpy(&t[i++],&a[p++],sizeof(a[0]));
            else
            {
                memcpy(&t[i],&a[q],sizeof(a[0]));
                c[a[q].i]+=m-p;//c[i]记录在i前面的比i大的个数
                i++;q++;
            }
        }
        for(int i=l;i<r;i++) memcpy(&a[i],&t[i],sizeof(a[0]));
    }
}

int main()
{
    //freopen("/home/user/in","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=0;i<n;i++)
        {
            a[i].i=i;
            scanf("%d",&a[i].x);
        }
        memset(c,0,sizeof(c[0])*(n+1));
        merge_sort(0,n);
        long long sum=0;
        //for(int i=0;i<n;i++) printf("a[%d]=%d %d %d
",i,a[i].x,c[a[i].i],a[i].i-c[a[i].i]);
        for(int i=0;i<n;i++)
            sum+=c[a[i].i]*(i-a[i].i+c[a[i].i])+(a[i].i-c[a[i].i])*(n-i-1-c[a[i].i]);
        printf("%lld
",sum); }
    return 0;
}

用分治法求逆序对的方法做的