其他分享
首页 > 其他分享> > 牛客线段树/树状数组练习

牛客线段树/树状数组练习

作者:互联网

NC14522 珂朵莉的数列(int128+树状数组)

题意:

珂朵莉有一个序列,有n*(n-1)/2个子区间,求出每个子区间逆序对的对数

思路:

如果是只求整个序列的逆序对个数的话,直接从后向前树状数组动态加点统计当前有多少个小于当前点的数就好了

但是要算所有区间的对数的话,暴力肯定会超时,所以我们可以计算每个逆序对对答案的贡献,也就是这个逆序对所在区间的个数,这个区间的个数很明显就为i*∑(n-j+1),其中j为i右边比它小的位置

然后树状数组下标存的就是(n-j+1),由于每个数出现的次数不一定为1,所以有的下标对应的值要累加,并且由于ai很大,要先进行离散化

还有比较坑的就是。。。最后居然连unsigned long long都爆了,要用int128

#include<iostream>
#include<algorithm>
#define lowbit(x) x&(-x)
 using namespace std;
 typedef unsigned long long ll;
 const int maxn=1e6+10;
 ll a[maxn],c[maxn],n,b[maxn];
inline void print(__int128 x){
    if(x<0){
        putchar('-');
        x=-x;
    }
    if(x>9)
        print(x/10);
    putchar(x%10+'0');
}
 void add(ll x,ll val)
 {
     while(x<=n){
         c[x]+=val;
         x+=lowbit(x);
     }
 }
 ll sum(ll x)
 {
     ll ans=0;
     while(x>=1){
         ans+=c[x];
         x-=lowbit(x);
     }
    return ans;
 }
 int main()
 {
     scanf("%lld",&n);
     __int128 ans=0;
     for(int i=1;i<=n;i++){
         scanf("%lld",&a[i]);
         b[i]=a[i];
     } 
    sort(b+1,b+1+n);
    int len=unique(b+1,b+1+n)-b-1;
    for(int i=n;i>=1;i--){
        int pos=lower_bound(b+1,b+1+len,a[i])-b;
        ans+=(__int128)i*sum(pos-1);
        add(pos,n-i+1);
    }
    print(ans);
    return 0;
 }
 
View Code

 

标签:树状,int,线段,long,牛客,ans,int128,ll,逆序
来源: https://www.cnblogs.com/overrate-wsj/p/12663154.html