P1908 逆序对 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
思路:维护一个线段树,按序列顺序,查找完ai就将它加入线段树,这样每次查找在线段树中比ai大的数就符合逆序对的要求。但序列中的数字最大有1e9,不能直接用线段树存,所以想先用离散化思想,用相对大小关系表示绝对大小,比如在题目中:1、2、3与1、2、99性质是一样的
#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define NOTLE ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define endl '\n'
#define T int TT;cin >> TT;while (TT--)
#define lint long long
#define pb push_back
lint a[500010],fa[500010],b[500010],c[500010];
lint tree[2000010]={0};
lint maxn=0,ans=0,x;
int bs(lint ob,int l,int r){ //二分查找实现离散化
while(l<r){
int mid=(l+r+1)/2;
if(ob>=b[mid]) l=mid;
else r=mid-1;
}
return r;
}
void update(lint i,int l,int r){
if(l==r){tree[i]++; return ;}
int mid=(l+r)/2;
if(x<=mid) update(i*2,l,mid);
else update(i*2+1,mid+1,r);
tree[i]=tree[i*2]+tree[i*2+1];
}
lint query(lint i,int l,int r){
if(x<=l && maxn>=r) return tree[i];
int mid=(l+r)/2;
lint add=0;
if(mid>=x) add+=query(i*2,l,mid);
if(mid<maxn) add+=query(i*2+1,mid+1,r);
return add;
}
int main(){
NOTLE;
int n;
cin >> n;
for(int i=1;i<=n;i++){
cin >> a[i];
fa[i]=a[i];
}
sort(fa+1,fa+1+n);
//离散化,用相对大小关系表示绝对大小
int cnt=1;
b[cnt]=fa[1];
for(int i=2;i<=n;i++)
if(fa[i]!=fa[i-1]) b[++cnt]=fa[i];
for(int i=1;i<=n;i++){
c[i]=bs(a[i],1,cnt+1);
maxn=max(maxn,c[i]); //找出最大值作为线段树右边
}
for(int i=1;i<=n;i++){
x=c[i]+1;
//查找范围c[i]+1~maxn,在这个范围内只要有数,就符合逆序对的要求
ans+=query(1,0,maxn);
x=c[i];
update(1,0,maxn);//将c[i]加入线段树
}
cout << ans;
return 0;
}