蒟蒻の疑惑

P1908 逆序对

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生成的


|