HDOJ 3743 Frosh Week(树状数组求逆序对)


http://acm.hdu.edu.cn/showproblem.php?pid=3743


这道题就是给出几个数排成一列之后,求逆序对的数量。


把所有的数从第一个开始,给另外一个数组下标对应为这个数的元素加1。假设当前刚刚把下标为x的元素加1,那么目前数组里有的数除了刚才那个数之外,就是排在它前面的所有数。下标为x以前的数都是比x小,对应的元素值就是出现的次数。由此每放进去一个数,都统计一下它前面放进去的数中有几个比它小,所有累加起来的值就是逆序对数。

再把普通一维数组转换成树状数组即可。

由于输入的数可能会跨度大,所以将数据离散化(只适用于只关注数据相对大小,不关注具体值的情况),处理方式就是先赋予这些数一个id,按照本身从小到大的顺序把这些数排好,从1开始逐个赋值,遇到相邻的数一样的情况也赋一样的数。再按id从小到大排一遍恢复原来的顺序。


#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 1000005
const int INF = 0x3f3f3f3f;
using namespace std;
int n, c[N];
struct Node{
int id;
int num;
};
struct Node a[N];
__int64 ans;
bool cmp(Node x, Node y)
{
    return x.num < y.num;
}
bool cmp2(Node x, Node y)
{
    return x.id < y.id;
}
int lowbit(int x)
{
    return x & (-x);
}
int query(int m)
{
    int sum = 0;
    while (m > 0)
    {
        sum += c[m];
        m -= lowbit(m);
    }
    return sum;
}
void add(int i, int x)
{
    while (i <= n)
    {
        c[i] += x;
        i += lowbit(i);
    }
}
int main()
{
    while(scanf("%d", &n) != EOF)
    {
        memset(c, 0, sizeof(c));
        for (int i = 1; i <= n; i++)
        {
            scanf("%d", &a[i].num);
            a[i].id = i;
        }
        sort(a+1, a+n+1, cmp);
        a[0].num = INF;
        for (int i = 1; i <= n; i++)
            if (a[i].num != a[i-1].num) a[i].num = i;
            else a[i].num = a[i-1].num;
        sort(a+1, a+n+1, cmp2);
        ans = 0;
        for (int i = 1; i <= n; i++)
        {
            add(a[i].num, 1);
            ans += query(n) - query(a[i].num);
        }
        printf("%I64d\n", ans);
   }
    return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值