求逆序对的和
分析:其实这个结果和逆序数有关,对某个位置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);
}
}