小奇遐想
小奇望着青天中的悠悠白云,开始了无限的遐想,在它的视野中,恰好有n朵高度不同的白云排成一排,他想从左到右选出四朵白云a,b,c,d,使得h_a<h_b<h_d<h_c,即看起来像是彩虹的形状!它想知道有多少种方案数。
输入
第一行包括1个整数n。
第二行包括n个整数,第i个正数表示h_i,保证这n个整数是n的一个全排列。
输出
输出一个整数表示答案。(mod 16777216)
样例输入
5
1 5 3 2 4
样例输出
0
提示
对于10%的数据n<=600;对于40%的数据n<=5000;
对于100%的数据n<=200000。
思路:
根据题意是要找形如1243的排列个数,那么我们可以先找出12XX和1234的排列个数,两者相减就是答案
代码:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod=16777216;
ll sum[200005],l[200005],r[200005],a[200005];
int n;
ll Ans1=0,Ans2=0;
void add(int i,ll m ) { while(i<=n) sum[i]+=m,i+=(-i&i); }
ll ask(int i){ ll ans=0; while(i>0) ans+=sum[i],i-=(-i&i); return ans;}
void LR()
{
for(int i=1;i<=n;i++)
{
l[i]=ask(a[i]);//i前面 小于i的个数
r[i]=n-i-(a[i]-l[i]-1);//i后面大于i的个数
add(a[i],1);
}
}
void _12XX()
{
for(int i=1;i<=n;i++)
{
Ans1+=l[i]*r[i]*(r[i]-1)/2;//枚举2位置
Ans1%=mod;
}
}
void _1234()
{
for(int i=1;i<=n;i++)
{
Ans2+=ask(a[i])*r[i];//枚举3位置
Ans2%=mod;
add(a[i],l[i]);//记录a[i]前小于a[i]的个数
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
memset(sum,0,sizeof(sum));
LR();
//for(int i=1;i<=n;i++)
//printf("%lld %lld\n",l[i],r[i]);
_12XX();
memset(sum,0,sizeof(sum));
_1234();
// printf("%lld %lld\n",Ans1,Ans2);
printf("%lld",(Ans1-Ans2+mod)%mod);
return 0;
}