逆序对(权值线段树)

题目
在这里插入图片描述
逆序对可以用归并和树状数组求,应该比线段树快,这里只是为了学习下权值线段树。
权值线段树就是维护每个数出现的次数,可以用来求数组的第K大元素。
而求逆序对个数,就是求每个数前面有多少大于自己的数,所以可以用权值线段树来写,
比如查询一个数a,
a<=mid,说明a在左子树中,所以所有柚子树的元素都比a大,答案就为左子树中大于a的数加上柚子树的数的个数。
否则a在柚子树中(什么傻逼输入法啊),答案为柚子树大于a的个数。

代码

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N = 6e5 + 7;
const int M = 2e3+7;
const int inf = 0x3f3f3f3f;
const int mod = 1e9+7;
typedef long long ll;
struct node
{
    ll sum;
    int l,r;
}tree[N<<2];
void build(int node,int l,int r)
{
    tree[node].l=l;
    tree[node].r=r;
    int mid=(l+r)>>1;
    if(l==r)return;
    build(node<<1,l,mid);
    build(node<<1|1,mid+1,r);
}
void update(int node,int val)
{
    if(tree[node].l==tree[node].r){
        tree[node].sum++;
        return;
    }
    int mid=(tree[node].l+tree[node].r)>>1;
    if(val<=mid)update(node<<1,val);
    else update(node<<1|1,val);
    tree[node].sum=tree[node<<1].sum+tree[node<<1|1].sum;
}
ll query(int node,int val)
{
    if(tree[node].l==tree[node].r)return tree[node].sum;
    int mid=(tree[node].l+tree[node].r)>>1;
    if(val<=mid)return query(node<<1,val)+tree[node<<1|1].sum;
    else return query(node<<1|1,val);
}
struct num
{
    int data,id;
    bool operator<(const num&t1)const{
        return data<t1.data;
    }
}a[N];
int b[N];
int main()
{
    int n;
    scanf("%d",&n);
    rep(i,1,n){
        scanf("%lld",&a[i].data);
        a[i].id=i;
    }
    sort(a+1,a+1+n);
    int cnt=0;
    a[0].data=-1;
    rep(i,1,n){
        if(a[i].data==a[i-1].data)b[a[i].id]=cnt;
        else b[a[i].id]=++cnt;
    }
    build(1,1,cnt);
    ll ans=0;
    rep(i,1,n){
        ans+=query(1,b[i]+1);
        update(1,b[i]);
    }
    printf("%lld\n",ans);
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值