题目链接
1.利用等差数列转化为求逆序对和正序对
2.树状数组优化
因为ans的输出问题WA了好多次。。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int N=1000010;
int n,a[N],c[N],l[N],r[N];
int ask(int x){
int res=0;
while(x>0){
res+=c[x];
x-=x&(-x);
}
return res;
}
void add(ll x,ll y){
while(x<=N){
c[x]+=y;
x+=x&(-x);
}
}
int main(){
cin>>n;
for(int i=1;i<=n;i++){
scanf("%lld",&a[i]);
a[i]++;
}
memset(c,0,sizeof(c));
for(int i=n;i;i--){
r[i]=ask(a[i]-1);
add(a[i],1);
}
memset(c,0,sizeof(c));
for(int i=1;i<=n;i++){
l[i]=ask(a[i]-1);
add(a[i],1);
}
/* for(int i=1;i<=n;i++){
cout<<r[i]<<endl;
}*/
ll ans=0;
for(int i=1;i<=n;i++){
ll t=(i-1-l[i])+r[i];
ans+=t*(t+1)/2;
}
printf("%lld\n",ans); //bug...
}