USACO 2017 February Contest Gold T3: Why Did the Cow Cross the Road III

题目大意

给定长度为2N的序列，1~N各出现过2次，i第一次出现位置记为ai，第二次记为bi，求满足 ai<aj<bi<bj 的对数

题目分析

把所有数对找出来，然后按左端点排序。令当前数对为j, 那么排序保证了 a_i<a_j<b_j ,然后每次统计左右端点之间的标记个数，即找出a_j与b_j间的b_i, 其个数就是该数对的答案。用树状数组维护即可。

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e5+10;


struct Node{
    int l,r;
}s[MAXN];
inline bool cmp(Node x,Node y){
    return x.l<y.l;
}

int n,ans;
int BIT[MAXN];
inline int lowbit(int x){
    return x&(-x);
}
inline void Update(int x,int v){
    for(int i=x;i<=n*2;i+=lowbit(i))
        BIT[i]+=v;
} 
inline int Query(int x){
    int res=0;
    for(int i=x;i;i-=lowbit(i))
        res+=BIT[i];
    return res;
}
int main(){
    scanf("%d",&n);
    for(int i=1,x;i<=n*2;++i){
        scanf("%d",&x);
        if(!s[x].l) s[x].l=i;
        else s[x].r=i;
    }
    sort(s+1,s+n+1,cmp);
    for(int i=1;i<=n;++i){
        Update(s[i].r,1);
        ans+=Query(s[i].r-1)-Query(s[i].l);
    }
    printf("%d
",ans);
    return 0;
}