C83 树状数组 P1908 逆序对

视频链接：269 树状数组 P1908 逆序对_哔哩哔哩_bilibili

Luogu P1908 逆序对

// 树状数组 点修+点查 O(nlogn)
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

#define N 500005
struct node{
  int val,pos; //值,位置
}a[N];
int n,s[N];
long long ans;

void add(int x,int k){ //点修
  while(x<=n) s[x]+=k, x+=x&-x;
}
int sum(int x){ //前缀和
  int t=0;
  while(x) t+=s[x], x-=x&-x;
  return t;
}
int cmp(node x,node y){ //降序排序
  if(x.val==y.val) return x.pos>y.pos; 
  return x.val>y.val;
}
int main(){
  scanf("%d",&n);
  for(int i=1;i<=n;i++) 
    scanf("%d",&a[i].val),a[i].pos=i;
  sort(a+1,a+n+1,cmp);
  for(int i=1;i<=n;i++){ //从前面找>ai的个数
    ans+=sum(a[i].pos-1);
    add(a[i].pos,1);
  }
  printf("%lld\n",ans);
}

// 离散化+权值树状数组 O(nlogn)
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

#define N 500005
int n,a[N],b[N],s[N];
long long ans;

void add(int x,int k){ //点修
  while(x<=n) s[x]+=k, x+=x&-x;
}
int sum(int x){ //前缀和
  int t=0;
  while(x) t+=s[x], x-=x&-x;
  return t;
}
int main(){
  scanf("%d",&n);
  for(int i=1;i<=n;i++) 
    scanf("%d",&a[i]),b[i]=a[i];
  sort(b+1,b+n+1);
  for(int i=1;i<=n;i++) //离散化
    a[i]=lower_bound(b+1,b+n+1,a[i])-b;
  for(int i=n;i;i--){ //从后面找<ai的个数
    ans+=sum(a[i]-1);
    add(a[i],1);
  }
  printf("%lld\n",ans);
}