树状数组求逆序对数

18 篇文章 0 订阅
5 篇文章 0 订阅

再清楚地打一边用树状数组求逆序对,记不住就只能多打,万事唯手熟耳。

这里都用的int,实际题目数据要求会不一样,必要时用long long int。这里的一步离散化的作用是如果数列中元素的大小比较大就不易直接开大小范围的树状数组来做,而是先按元素值排序然后比较的是元素原来的index,道理是一样的。

#include <iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
class node//节点
{
public:
    int value;
    int index;
};

node nodes[100005];//输入数列
int a[100005];//离散化数组
int c[100005];//树状数组
int n;//数列长度
int result;//结果值
bool cmp(node a,node b)//按节点值比较,用来排序
{
    if(a.value<b.value)
        return true;
    else return false;
}

int lowbit(int x)//lowbit函数
{
    return x&(-x);
}

void update(int t,int value)//更新树状数组或叫插入元素,有地方叫change或add
{
    for(int i=t;i<=n;i+=lowbit(i))
    {
        c[i]+=value;
    }
}

int getsum(int x)//求前几项和,这里用来求x之前比x大的个数
{
    int ans=0;
    for(int i=x;i>=1;i-=lowbit(i))
    {
        ans+=c[i];
    }
    return ans;
}



int main()
{
    scanf("%d",&n);
    int tmp;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&tmp);
        nodes[i].value=tmp;
        nodes[i].index=i;
    }

    sort(nodes+1,nodes+n+1,cmp);//按节点的value排序

    for(int i=1;i<=n;i++)//离散化
    {
        a[nodes[i].index]=i;
    }

    result=0;
    for(int i=1;i<=n;i++)
    {
        update(a[i],1);
        result+=i-getsum(a[i]);//之前比a[i]大的数
    }

    printf("%d\n",result);
    return 0;
}


 

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值