C83 树状数组 P1908 逆序对
视频链接:269 树状数组 P1908 逆序对_哔哩哔哩_bilibili
// 树状数组 点修+点查 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); }
同题