P3697乒乓比赛 | ||
|
问题描述
n个乒乓球爱好者站成一排,从左往右编号1到n。他们要组织比赛切磋技术。每个人都有一个不同的技能值,编号i个人的技能值为ai。
每场比赛需要3个人:两名选手,一名裁判。要求作为裁判的那一个乒乓球爱好者所站的位置必须在两名选手的之间,并且技能值也在两名选手的之间。
问一共能组织多少场不同的比赛。
输入格式
第一行,一个整数n。
第二行,n个空格间隔的整数,按照从左往右的顺序给出了每个乒乓球爱好者的能力值ai。
输出格式
一行,一个整数,表示所求结果
样例输入 1
3
1 2 3
样例输出 1
1
样例输入 2
5
9 1 4 3 6
样例输出 2
3
提示
3<=n<=50000
1<=ai<=200000
来源 改编自LA 4329
考虑当前第i个人当裁判的情况,则第i个人当裁判能得到的比赛场数等于左边能力值比他小的人数×右边能力值比他大的人数+左边能力值比他大的人数×右边能力值比他小的人数
而各个方向的能力值的求法如下:
用s[i]记录第i个人的能力,c[x]为树状数组,c数组对应的a[x]数组代表数x是否出现,如果出现a[x]=1否则为0
#include<cstdio>
#include<iostream>
#include<cstring>
#define LL long long
#define lowbit(x) x&(-x)
using namespace std;
const LL maxn1=50002;
const LL maxn2=200002;
LL n,x,a[maxn1];
LL c[maxn2],ls[maxn2],rs[maxn2];//ls为左边小于,rs为右边大于
void _read(LL &x){
char tt=getchar();
while(tt<'0'||'9'<tt) tt=getchar();
for(x=0;'0'<=tt&&tt<='9';x=x*10+tt-'0',tt=getchar());
}
void modify(LL x,LL d){for(LL i=x;i<=maxn2;i+=lowbit(i))c[i]+=d;}
LL getsum(LL x){
LL sum=0;
for(LL i=x;i;i-=lowbit(i))sum+=c[i];
return sum;
}
int main(){
LL i,ans=0;
_read(n);
for(i=1;i<=n;i++)_read(a[i]);
modify(a[1],1);
for(i=2;i<=n;i++){
ls[i]=getsum(a[i]-1);
modify(a[i],1);//先讨论再插入数据,能保证讨论到的数据都出现在当前讨论的人的左边
}
memset(c,0,sizeof(c));//要重新插入,故先清空数组
modify(a[n],1);
for(i=n-1;i;i--){
rs[i]=getsum(maxn2)-getsum(a[i]);
modify(a[i],1);
}
for(i=1;i<=n;i++)
ans+=ls[i]*rs[i]+(i-1-ls[i])*(n-i-rs[i]);
cout<<ans;
}