hdu 2838 cow sorting(树状数组)

求逆序对的和

分析:其实这个结果和逆序数有关,对某个位置i,如果前面比他大的有x个,那么a[i]至少要加x次 
如果后面有y个比a[i]小,那么a[i]至少要加y次,也就是说用两个树状数组来分别维护当前位置时前面有多少比他大,后面有多少个比他小

求逆序数两两的总和: 如 3 2 1 :sum=(3+2)+(3+1)+(2+1)=12;   1 2 3: sum=0;

   对于新插入的一个元素,运用树状数组,可以求得比它小的元素的个数,比它小的元素的和,在它之前的元素的总和。

    而对于每一个新元素,其sum[m]=m*(比它大的元素个数)+(前i个元素的和)-(比它小的元素的和)。

    然后累加得解。


可得出公式

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

typedef  __int64 ll;

struct node {
   int id;
   ll sum;
}s[100005];

int n;
//更新个个节点
void add(int i,int j,int k){
	while(i<=n){
      s[i].id+=j;
      s[i].sum+=k;
      i+=i&-i;
	}
}
//此函数求逆序对的个数
int get_id(int i){
	int idd=0;
	while(i>0){
    idd+=s[i].id;
    i-=i&-i;
	}
	return idd;
}
//求前i个数的和
ll get_sum(int i){
      ll summ=0;
      while(i>0){
      	summ+=s[i].sum;
      	i-=i&-i;
      }
      return summ;
}

int main(){
    while(~scanf("%d",&n))
    {
        int x;
        memset(s,0,sizeof(s));
        ll ans=0,k;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);
            add(x,1,x);
           ll k=(i-get_id(x));
            if(k==0)
                continue;
            ans+=(k*x+get_sum(n)-get_sum(x));
        }
        printf("%I64d\n",ans);
    }
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值