codeforces 540E Infinite Inversions(离散化+树状数组)

题目:一个无限数列,1,2,3,4,...,....,给n个数对把数列的i,j两个位置的元素交换。求交换后数列的逆序对数。

 

思路:离散点并且把两点之间的m个数当作一个数来看记下贡献,然后就是普通的树状数组求逆序对,需要用long long!

#include 

using namespace std;
typedef long long ll;
const int maxn=2e5+10;
int l[maxn],r[maxn],num[maxn<<1],b[maxn],a[maxn<<1],c[maxn<<1],n,cnt;
void update(int pos,int val)
{
    for(int i=pos;i<=cnt;i+=i&-i)
        c[i]+=val;
}
int query(int pos)
{
    int ans=0;
    for(int i=pos;i>0;i-=i&-i)
        ans+=c[i];
    return ans;
}
int main()
{
    scanf("%d",&n);
    cnt=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&l[i],&r[i]);
        b[++cnt]=l[i];b[++cnt]=r[i];
    }
    sort(b+1,b+cnt+1);
    int o=unique(b+1,b+cnt+1)-b-1;
    cnt=0;
    for(int i=1;i<=o;i++)
    {
        if(b[i-1]+1!=b[i])
        {
            a[++cnt]=b[i-1]+1;
            num[cnt]=b[i]-b[i-1]-1;
        }
        a[++cnt]=b[i];
        num[cnt]=1;
    }
    for(int i=1;i<=n;i++)
    {
        l[i]=lower_bound(a+1,a+cnt+1,l[i])-a;
        r[i]=lower_bound(a+1,a+cnt+1,r[i])-a;
    }
    for(int i=1;i<=cnt;i++)
        a[i]=i;
    for(int i=1;i<=n;i++)
        swap(a[l[i]],a[r[i]]);
    ll ans=0;
    for(int i=1;i<=cnt;i++)
    {
        ans+=1ll*(query(cnt)-query(a[i]))*(num[a[i]]);
        update(a[i],num[a[i]]);
    }
    printf("%lld\n",ans);
    return 0;
}

 

你可能感兴趣的:(树状数组)