Wuzke_ @ 2024-10-20 21:14:46
#include<iostream>
#include<algorithm>
#include<limits.h>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#define int long long
#define endl '\n'
using namespace std;
struct node{
int val,pos;
}a[1145141];
int n,ans=0;
int tree[1145141];
int lowbit(int x){
return x&-x;
}
void add(int x,int k){
for(;x<=n;x+=lowbit(x)) tree[x]+=k;
}
int ask(int x){
int ans=0;
for(;x;x-=lowbit(x)) ans+=tree[x];
return ans;
}
bool cmp(node x,node y){
if(x.val!=y.val){
return x.val<y.val;
}
return x.pos<y.pos;
}
signed main(){
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
cin>>n;
memset(tree,0,sizeof(tree));
for(int i=1;i<=n;i++){
cin>>a[i].val;
a[i].pos=i;
}
stable_sort(a+1,a+n+1,cmp);
//--------------------------
for(int i=n;i>=1;i--){
ans+=ask(a[i].pos);
add(a[i].pos,1);
}
//--------------------------
cout<<ans<<endl;
return 0;
}
看着题解区dalao的代码自己敲了一遍
发现两条分割线内的代码还是不懂qwq
望得到解答,感谢
by Dino21 @ 2024-11-16 11:16:49
@Wuzke_
•这段代码的核心是计算逆序对的数量。
•从最后一个元素开始遍历,对于每个元素 a[i],使用 ask(a[i].pos) 计算在 a[i].pos 之前已经出现的元素数量,这些元素的值一定大于 a[i].val,因此它们构成逆序对。
•然后使用 add(a[i].pos, 1) 将 a[i].pos 位置的计数加 1,以便后续的元素可以正确地计算逆序对。输出结果cout << ans << endl; return 0;
•最后输出逆序对的数量 ans。总结这段代码的主要目的是计算一个数组中的逆序对数量。通过使用树状数组(BIT),可以在 O(n log n) 时间复杂度内高效地完成这一任务。关键在于理解 ask 和 add 函数的作用,以及如何利用它们来计算逆序对。
总结这段代码的主要目的是计算一个数组中的逆序对数量。通过使用树状数组(BIT),可以在 O(n log n) 时间复杂度内高效地完成这一任务。关键在于理解 ask 和 add 函数的作用,以及如何利用它们来计算逆序对。
求关
by Wuzke_ @ 2024-11-16 22:40:19
@Dino21
一眼盯真
AI生成的