#pragma GCC optimize ("Ofast")
#pragma GCC optimize ("unroll-loops")
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<unordered_map>
using namespace std;
#define int long long
const int N = 1000010;
unordered_map<int,int> mp;
int n, a[N];
vector<int> vec;
int tr[N];
int lowbit(int x)
{
return x & -x;
}
void add(int x,int v)
{
for(int i = x; i < N; i += lowbit(i)) tr[i] += v;
}
int query(int x)
{
int res=0;
for(int i=x;i>=1;i-=lowbit(i))res+=tr[i];
return res;
}
int get(int x)
{
return lower_bound(vec.begin(),vec.end(),x)-vec.begin()+1;
}
int maxv[N], minv[N];
void solve()
{
cin >> n;
for(int i = 1; i <= n; i ++) cin >> a[i], vec.push_back(a[i]);
sort(vec.begin(),vec.end());
vec.erase(unique(vec.begin(),vec.end()),vec.end());
for(int i = 1; i <= n; i ++) mp[get(a[i])] ++;
for(int i = 0; i < N; i ++) tr[i] = 0;
for(int i = 1; i <= n; i ++)
{
maxv[i] = query(N - 1) - query(get(a[i]));
add(get(a[i]), 1);
}
for(int i = 0; i < N; i ++) tr[i] = 0;
for(int i = n; i >= 1; i --)
{
minv[i] = query(get(a[i]) - 1);
add(get(a[i]), 1);
}
int res = 0;
for(int i = 1; i <= n; i ++)
res += maxv[i] * minv[i];
cout << res << '\n';
}
signed main()
{
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
int T; T = 1;
while(T --) solve();
}
三元组 ---树状数组 + 离散化 + 权值相乘
最新推荐文章于 2024-07-15 21:33:54 发布