这道题是08年北京现场赛的题。
讲一下题意,N个人,从中选出3个人,使得,这三个人必须是从大到小或者从小到大,并且是严格的大小关系,即不可相等。问有多少种这样的组合。
这道题的结论很好得出,假设一个人位于i,那么我们只需求出求1 -> (i - 1)位置小于这个人的个数,记为x ,当然大于这个人的个数也就出来了,记为x',同理, 求出(i + 1) -> n位置小于这个人的个数,记为y ,那么大于他的个数即为y'。
那么最后的答案就是sigam(x * y' + x' * y) 。从1 -> n累加即可。
那么问题就转化成求一个数他前面有多少个数小于他,后面有多少个数小于他。
对于这个问题,我们显然可以用树状数组来求解,对于求前缀的时候,可以从1 -> n这样插入,然后每次插入都求一次前缀和即可。
同理对于求后面的个数,可以从n -> 1这样插,然后同样每次求前缀和即可。
当然因为必须是严格的大小关系,每次求和的时候必须是query(a[i] - 1),这样就可以了。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <stack>
#include <map>
#include <iomanip>
#define PI acos(-1.0)
#define Max 2005
#define inf 2000000000
#define LL(x) (x<<1)
#define RR(x) (x<<1|1)
#define REP(i,s,t) for(int i=(s);i<=(t);++i)
#define ll long long
#define mem(a,b) memset(a,b,sizeof(a))
#define mp(a,b) make_pair(a,b)
using namespace std;
#define N 222222
int c[N] ;
int a[N] ;
int pre[N] ;
int suf[N] ;
int n , m ;
int lowbit(int x){return x & (-x) ;}
void updata(int x ){
for (int i = x ; i <= m ; i += lowbit(i))c[i] ++ ;
}
int query(int x){
int ans = 0 ;
for (int i = x ; i >= 1 ; i -= lowbit(i)){
ans += c[i] ;
}
return ans ;
}
void init(){
mem(c ,0) ;
mem(pre ,0) ;
mem(suf, 0) ;
}
void solve(){
cin >> n ;
init() ;
m = 0 ;
for (int i = 1 ; i <= n ; i ++ ){
scanf("%d",&a[i]) ;
m = max(a[i] , m) ;
}
for (int i = 1 ; i <= n ; i ++ ){
updata(a[i]) ;
pre[i] = query(a[i] - 1) ;
}
mem(c ,0) ;
for (int i = n ; i >= 1 ; i -- ){
updata(a[i]) ;
suf[i] = query(a[i] - 1) ;
}
ll ans = 0 ;
for (int i = 1 ; i <= n ; i ++ ){
ans += (ll)pre[i] * (n - i - suf[i]) + (ll)(i - pre[i] - 1) * suf[i] ;
}
cout << ans << endl;
}
int main() {
int _ ;cin >> _ ;while(_ -- ) solve() ;
return 0;
}