小朋友排队(树状数组)详细注释

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
//分析可知每个小朋友的交换次数即每个数左边更大和右边更小的数的个数和,即左边与其形成的逆序对数 + 右边与其形成的逆序对数
//则此问题转换为求逆序对数,如果直接暴力求解时间复杂度o(n^2)会超限,则利用树状数组
struct node
{
    int val;    //值
    int pos;    //输入次序
}a[100001];
int cmp(const node &a,const node &b)
{
    return a.val<b.val;
}
int cmp1(const node &a,const node &b)
{
    return a.pos<b.pos;
}


int n;
int c[100001]={0};
int lowbit(int x)
{
    return x&(-x);
}
void update(int x)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        c[i]+=1;
    }
}
int getsum(int x)
{
    int sum=0;
    for(int i=x;i>0;i-=lowbit(i))
    {
        sum+=c[i];
    }
    return sum;
}
//由题0<=Hi<=1000000,如果直接建立树状数组会消耗大量空间,可以先把输入的数据变成从1到n,从而节省空间
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i].val;
        a[i].pos=i;     //记录输入次序
    }
    sort(a+1,a+n+1,cmp);
    int no=1;
    int temp;
    for(int i=1;i<=n;i++)   //把输入的数据变成从1到n,注意重复数的处理
    {
        if(a[i].val!=temp&&i!=1)
            no++;
        temp=a[i].val;
        a[i].val=no;
    }
    sort(a+1,a+1+n,cmp1);
    long long int b[100001]={0};  //注意int可能会超限
    for(int i=1;i<=n;i++)      //计算每个数左边与其形成的逆序对数
    {
        update(a[i].val);
        b[i]=i-getsum(a[i].val);
    }
    memset(c,0,sizeof(c));
    for(int i=n;i>0;i--)         //计算每个数右边与其形成的逆序对数
    {
        update(a[i].val);
        b[i]+=getsum(a[i].val-1);
    }
    unsigned long long int sum=0;  //注意int可能会超限
    for(int i=1;i<=n;i++)
    {
        sum+=((1+b[i])*b[i])/2;
    }
    cout<<sum;

}

树状数组学习参考博客:https://www.cnblogs.com/hsd-/p/6139376.html

https://blog.csdn.net/queuelovestack/article/details/52180767

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值