与树状数组与逆序对的想法一样。 具体可以看博客
维护出每一个位置,左边有多少比它大的(n),右边有多少比它大的(m),这样可以知道,以这个点为最低点构成的v的数量。
维护操作: 统计左边有多少比它大的(n)。计算区间[a[i]+1,n]中出现元素的个数,更新,当前点的值为1. 表示这个点出现次数加1.
其它维护也类似。
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include<bits/stdc++.h>
#define int long long
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const double eps = 1e-5;
const int mod = 998244353;
const int N = 2e5+10;
int A[N],V[N],c[N];
int n,a[N],ansA,ansV;
int lowbit(int x) {return x & -x;}
void update(int x,int y){
for(int i=x;i<=n;i+=lowbit(i)) c[i] += y;
}
int ask(int x){
int sum = 0;
for(int i=x;i;i-=lowbit(i)) sum += c[i];
return sum;
}
signed main(){
IOS
#ifdef ddgo
freopen("C:\\Users\\asus\\Desktop\\ddgoin.txt","r",stdin);
#endif
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=n;i>=1;i--){
V[i] = ask(n) - ask(a[i]-1);
A[i] = ask(a[i]-1);
update(a[i],1);
}
memset(c,0,sizeof(c));
for(int i=1;i<=n;i++){
ansV += V[i] * (ask(n) - ask(a[i]-1));
ansA += A[i] * ask(a[i]-1);
update(a[i],1);
}
cout<<ansV<<" "<<ansA<<endl;
return 0;
}