树状数组求逆序对 + 离散化(poj2299)

最新推荐文章于 2023-01-23 23:58:43 发布

转载最新推荐文章于 2023-01-23 23:58:43 发布 · 316 阅读

数据结构基础专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种使用树状数组求解逆序对数量的方法，并通过离散化处理来应对输入数据范围过大的问题。文章提供了一个完整的C++实现示例，包括自定义结构体、比较函数、树状数组的增删查操作等。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

转载注明出处
https://www.cnblogs.com/xiongmao-cpp/p/5043340.html

今天做了一个树状数组求逆序对的题，需要离散化，看了博客明白了为什么要进行离散化，原因是树状数组中的C[MAXN]数组其实相当于一个哈希操作，如果所给数组中存在值特别大的元素，数组就需要离散化

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
struct A
{
    int i;
    int v;
};
const int MAXN = 5e5 + 10;
A a[MAXN];
int c[MAXN];
typedef long long LL;
int lowbit(int i)
{
    return i & (-i);
}
int n;
void add(int i,int v)
{
    while(i <= n)
    {
        c[i] += v;
        i += lowbit(i);
    }
    return ;
}
int query(int i)
{
    int ans = 0;
    while(i)
    {
        ans += c[i];
        i -= lowbit(i);
    }
    return ans;
}
bool cmp(A a,A b)
{
    return a.v < b.v;
}
int main()
{
    ios::sync_with_stdio(false);
    while(cin >> n && n)
    {
        memset(c,0,sizeof(c));
        for(int i = 1; i <= n; ++i)
        {
            cin >> a[i].v;
            a[i].i = i;
        }
        sort(a + 1,a + n + 1,cmp);
        LL ans = 0;
        for(int i = 1; i <= n; ++i)
        {
            add(a[i].i,1);
            ans += i - query(a[i].i);
        }
        cout << ans << endl;
    }
    return 0;
}