Infinite Inversions CodeForces - 540E

http://codeforces.com/problemset/problem/540/E

先把交换的位置离散 然后相邻两个位置之间的线段也拿出来 把点看作长度为1的线段 然后这些线段构成一个新的序列 线段长度作为权值 线段右端点作为序列中的值 然后跑一遍逆序对就好

#include <bits/stdc++.h>
using namespace std;
#define ll long long

struct node0
{
    int val;
    ll cnt;
};

struct node1
{
    int l;
    int r;
};

struct node2
{
    int l;
    int r;
    ll val;
};

node0 pre[400010];
node1 seg[100010];
node2 tree[1600010];
int tmp[400010],pos[400010];
int n,len,tot;

void build(int l,int r,int cur)
{
    int m;
    tree[cur].l=l,tree[cur].r=r,tree[cur].val=0;
    if(l==r) return;
    m=(l+r)/2;
    build(l,m,2*cur);
    build(m+1,r,2*cur+1);
}

ll query(int pl,int pr,int cur)
{
    ll res;
    if(pl<=tree[cur].l&&tree[cur].r<=pr) return tree[cur].val;
    res=0;
    if(pl<=tree[2*cur].r) res+=query(pl,pr,2*cur);
    if(pr>=tree[2*cur+1].l) res+=query(pl,pr,2*cur+1);
    return res;
}

void update(int tar,ll val,int cur)
{
    tree[cur].val+=val;
    if(tree[cur].l==tree[cur].r) return;
    if(tar<=tree[2*cur].r) update(tar,val,2*cur);
    else update(tar,val,2*cur+1);
}

int main()
{
    ll ans,res;
    int i,p;
    scanf("%d",&n);
    len=0;
    for(i=1;i<=n;i++)
    {
        scanf("%d%d",&seg[i].l,&seg[i].r);
        pos[++len]=seg[i].l,pos[++len]=seg[i].r;
    }

    sort(pos+1,pos+len+1);
    len=unique(pos+1,pos+len+1)-pos-1;
    for(i=1;i<=n;i++)
    {
        seg[i].l=lower_bound(pos+1,pos+len+1,seg[i].l)-pos;
        seg[i].r=lower_bound(pos+1,pos+len+1,seg[i].r)-pos;
    }
    for(i=1;i<=len;i++) tmp[i]=i;
    for(i=1;i<=n;i++) swap(tmp[seg[i].l],tmp[seg[i].r]);
    //printf("***%d***\n",len);
    //for(i=1;i<=len;i++) printf("%d\n",tmp[i]);

    tot=0;
    for(i=1;i<=len;i++)
    {
        if(pos[i-1]+1<=pos[i]-1)
        {
            tot++;
            pre[tot].val=pos[i]-1;
            pre[tot].cnt=(ll)((pos[i]-1)-(pos[i-1]+1)+1);
        }
        tot++;
        pre[tot].val=pos[tmp[i]];
        pre[tot].cnt=1ll;
    }
    for(i=1;i<=tot;i++) tmp[i]=pre[i].val;

    sort(tmp+1,tmp+tot+1);
    //printf("***%d***\n",tot);
    //for(i=1;i<=tot;i++) printf("%d\n",tmp[i]);

    build(1,tot,1);
    ans=0;
    for(i=1;i<=tot;i++)
    {
        p=lower_bound(tmp+1,tmp+tot+1,pre[i].val)-tmp;
        if(p<tot)
        {
            res=query(p+1,tot,1);
            ans+=pre[i].cnt*res;
        }
        update(p,pre[i].cnt,1);
    }
    printf("%lld\n",ans);
    return 0;
}