AcWing 1215. 小朋友排队
原题链接
中等
作者:
月亮事务所
,
2021-02-13 23:45:23
,
所有人可见
,
阅读 404
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N= 1000010;
int n;
int h[N], tr[N];
int sum[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; i -= lowbit(i)) res += tr[i];
return res;
}
int main()
{
scanf("%d",&n);
for(int i=O;i<n;i ++)scanf("%d",&h[i]);
for(int i=0;i<n;i++)
{
sum[i]=query(N-1)-query(h[i]);
add(h[i],1);
}
for(int i=O;i<n;i++)
cout << sum[i] <<"";
for(int i =n-1;i>=0;i--)
{
sum[i]+= query(h[i]-1);
add(h[i],1);
}
LL res=0;
for(int i=0;i<n;i++)
res+=(LL)sum[i]*(sum[i] +1)/ 2;
cout << res <<endl;
return 0;
}