树状数组求逆序对 poj2299

树状数组求逆序对;
对于一组任意顺序的数,记下他们的初位置,再进行排序,记下他们本来应该在的位置。然后,按最初位置从前往后,进行处理。
基本思想:
对于一个数要求它对应得逆序对,可以通过求前面小于等于它的数的个数,然后用当前已处理的数的个数-前面小于等于当前的数的个数得到。
如:
4 3 5 2 1
排序后 1 2 3 4 5
对于第一个数4,当前处理数的个数为1,本来应该在的位置是4,则在第4个位置上加1(利用树状数组的单点修改),然后用树状数组的区间查询,求出前4项的和1,用1-1=0。所以4对应得逆序数个数为0。
对于第二个数3,当前处理数的个数为2,本来应该在的位置是3,则在第3个位置上加1(利用树状数组的单点修改),然后用树状数组的区间查询,求出前3项的和1,用2-1=1。所以3对应得逆序数个数为1。
以此类推。
注意当数组中有重复的数字时,他们本来应该在的位置应记为同一个位置。
附上poj 2299 AC代码

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=5e5+5;
int n;
struct node
{
    int val;
    int num;
}b[N];
int a[N];
int tree[N];
bool cmp(node a,node b)
{
    return a.val<b.val;
}
void add(int p,int x)//树状数组的应用
{
    while(p<=n)
    {
        tree[p]+=x;
        p+=p&(-p);
    }
}
int read(int p)
{
    int sum=0;
    while(p>=1)
    {
        sum+=tree[p];
        p-=p&(-p);
    }
    return sum;
}
int main()
{
    while(scanf("%d",&n),n)
    {
        memset(tree,0,sizeof(tree));
        memset(b,0,sizeof(b));
        memset(a,0,sizeof(a));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&b[i].val);
            b[i].num=i;//记下原位置
        }
        sort(b+1,b+1+n,cmp);
        a[b[1].num]=1;
        int cnt=1;
        for(int i=2;i<=n;i++)//元素的重复,顺序排列后应在同一位置
        {
            if(b[i].val!=b[i-1].val)
                cnt++;
            a[b[i].num]=cnt;
        }
        long long ans=0;
        for(int i=1;i<=n;i++)
        {
            add(a[i],1);
            ans+=(i-read(a[i]));
        }
        printf("%lld\n",ans);//用long long ,int 不够--WA!
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值