线段树求逆序对（POJ2299）

最新推荐文章于 2022-10-04 02:19:01 发布

原创最新推荐文章于 2022-10-04 02:19:01 发布 · 413 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

题目同时被 3 个专栏收录

102 篇文章

订阅专栏

模板

44 篇文章

订阅专栏

树

8 篇文章

订阅专栏

这篇博客介绍了如何使用线段树解决POJ2299问题，即计算序列中的逆序对数量。通过建立哈希表并逐个加入数字，统计每个数字后面的更大数值之和，最终实现逆序对的高效计算。对于可能的大数情况，可以采用离散化的方法避免空间问题。

POJ2299

逆序对即给定一行序列，问对于每一个数字，位于该数字前面且大于该数字的共有多少个。
事实上，我们可以建立一个哈希表，把输入的数字全部统计进去，每计算一个数字的逆序数时，统计该数字后面的值之和，例如：
5 3 7 4 1
建立哈希表：m[1],m[3],m[4],m[5],m[7] 初始都是0
现在加入5，则m[5]++,统计m[5]后面的数（m[7]）之和为0
加入3，则m[3]++,统计m[3]后面的数（m[4]+m[5]+m[7]）之和为1
加入7，m[7]++,后和为0
加入4，m[4]++,m[5]+m[7]=2
加入1，m[1]++,m[3]+m[4]+m[5]+m[7]=4
算出每个数的逆序数为 0 1 0 2 4
算连续区间的和可以用线段树来维护
可能给出的数过大，会爆空间，可以提前离散化一下

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn=500500;

ll a[maxn],b[maxn];
ll sum[4*maxn]={0};
ll n;

void LSH()
{
    sort(b+1,b+n+1);
    for(int i=1;i<=n;i++)
    {
        a[i]=lower_bound(b+1,b+n+1,a[i])-b;
    }
}

void PushUp(int n)
{
    sum[n]=sum[n*2]+sum[n*2+1];
    return;
}

void Update(int x,int l,int r,int num)
{
    if(l==r)
    {
        sum[num]++;
        return;
    }
    int m=(l+r)/2;
    if(x<=m)
        Update(x,l,m,num*2);
    else
        Update(x,m+1,r,num*2+1);
    PushUp(num);
}

int Query(int L,int R,int l,int r,int num)
{
    if(L<=l&&R>=r)
    {
        return sum[num];
    }
    int m=(l+r)/2;
    int ans=0;
    if(L<=m)
        ans+=Query(L,R,l,m,num*2);
    if(R>=m+1)
        ans+=Query(L,R,m+1,r,num*2+1);
    return ans;
}

int main()
{
    while(~scanf("%lld",&n))
    {
        if(n==0)
            break;
        ll ans=0;
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&b[i]);
            a[i]=b[i];
        }
        LSH();
        n++;
        for(int i=1;i<=n-1;i++)
        {
            Update(a[i],1,n,1);
            ans+=Query(a[i]+1,n,1,n,1);
        }
        printf("%lld\n",ans);
    }
    return 0;
}