思路
以后但凡你能看到一道题可以:
修改某个数,或者求前缀和
那么都可以考虑使用树状数组
这道题的思路感觉总体上和求逆序对的感觉上差不多
总体上你可以在坐标轴上画一画,数出来构成V字形和倒V字形的3个数据的个数。
那么每一次我去找每一个数的左边比他大的乘上右边比它大的,然后求出所有的这样的和,就是我们要找到的V字形的组数
反之同理,那就找小的。
那么如何找就是一个问题了:
可以顺序遍历一遍,把在左边比它小的/大的分别用数组记录下来
然后再倒叙遍历一遍,把在右边比它小的/大的,和原数组相乘起来,乘积相加便是结果了。
#include<iostream>
#include<cstring>
using namespace std;
const int N=2e5+7;
typedef long long ll;
int tree[N];
int up[N];
int down[N];
int lowbit(int x){
return x&(-x);
}
void add(int p,int x){
for(int i=p;i<=N;i+=lowbit(i)){
tree[i]+=x;
}
return;
}
int query(int x){
int res=0;
for(int i=x;i;i-=lowbit(i)){
res+=tree[i];
}
return res;
}
int query(int x,int y){
return query(y)-query(x-1);
}
int a[N];
int main(){
int n;
cin>>n;
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
for(int i=1;i<=n;i++){
up[i]=query(a[i],n);
down[i]=query(a[i]);
add(a[i],1);
}
ll ans1=0,ans2=0;
memset(tree,0,sizeof tree);
for(int i=n;i;i--){
ans1+=up[i]*(ll)query(a[i],n);
ans2+=down[i]*(ll)query(a[i]);
add(a[i],1);
}
cout<<ans1<<' '<<ans2<<endl;
}