树状数组

P1908 逆序对

Celestinefly @ 2024-04-20 22:22:58

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;

const int N=5e5+10;
int n;
int a[N],s[N],rp[N];
int lowbit(int x){
    return x&(-x);
}

void change(int k,int x){
    while(x<=n){
        s[x]+=k;
        x+=lowbit(x);
    }
}

int query(int x){
    int t=0;
    while(x>=1){
        t+=s[x];
        x-=lowbit(x);
    }
    return t;
}

bool cmp(int x,int y){
    if(a[x]==a[y]) return x>y;
    return a[x]>a[y];
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        rp[i]=i;
    }
    int sum=0;
    sort(a+1,a+n+1,cmp);
    for(int i=1;i<=n;i++){
        change(1,rp[i]);
        sum+=query(rp[i]-1);
    }
    printf("%d\n",sum);
    return 0;
}

我这个连案例都过不了,谁能帮我看看啊


by haokee @ 2024-04-20 23:21:07

@Celestinefly 你的程序只对 a 进行了排序,rp 没有任何变化。修改很简单,就是将 a 和 rp 塞到一个结构体里面,排序的时候就可以对 a 和 rp 同时进行排序了。


|