题目 1439: 蓝桥杯历届试题-小朋友排队
旧题新做,总有不一样的感觉。
不只是会忘记,温故仍可知新。
很简单的题。
用两次树状数组即可。
需要求前面有多少个比它大的,后面有多少个比它小的。
前面比他大的就是逆序对。以往都是数据离散化求,这个题可以不用离散化,因为身高最大才1e6.求逆序对,前 i 个数(包括第i个数)可以和第i个数组成 i 个顺序对或者是逆序对,或者相等。所以,query(a[i]-1)是之前求逆序对,是因为query(a[i])也包含了自己即求了相等对。所以query(a[i])包括了自己,相等对。所以用所有的 i - 顺序对和相等对即逆序对。
下面把tree memset一下,逆序add a[i] ,看后面有多少个比他小的。就是直接求顺序对了,直接query(a[i] - 1)即可。
最后的结果保存到一个res数组,最后的结果 = (res[i] + 1) * res[i] / 2*1ll 因为相乘可能超过了int。
还有一点就是输入的身高是从0开始的,树状数组的下标是从0开始的,所有一开始输入的时候要把每个a[i] + 1
世界缤纷美丽。
稻香真的好听。
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 1000000+10;
int tree[N];
int a[N];
int res[N];
int n,m=0;
int lowbit(int x)
{
return x&-x;
}
void add(int x,int c)
{
for(int i=x;i<N;i+=lowbit(i))
tree[i]+=c;
}
int query(int x)
{
int ans = 0;
for(int i=x;i>0;i-=lowbit(i))
ans+=tree[i];
return ans;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
++a[i];
// m=max(m,a[i]);
add(a[i],1);
res[i]+=i-query(a[i]);
}
//for(int i=1;i<=n;i++) cout<<res[i]<<" ";cout<<"\n";
memset(tree,0,sizeof(tree));
for(int i=n;i>=1;i--)
{
add(a[i],1);
res[i] +=query(a[i] - 1);
}
long long ans = 0;
for(int i=1;i<=n;i++)
{
// cout<<res[i]<<" ";
ans += 1ll*(res[i]+1)*res[i]/2;
}
cout<<ans;
}