pingpong uva1428

题目

求 三元组(i,j,k), i<j<k, 满足 a[i]<a[j]<a[k] 或者 a[i]>a[j]>a[k]，有多少组？(a[i] <=1e5)

枚举 j , 考虑 a[i]<a[j] 有多少 i 满足这个条件 c[j]

正反求出 c[ i ] ,d[ i ]

ans= c[i]*(n-i-d[i]) + (i-c[i]-1])*d[i]
————————————————

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
 
 const int M=1e5+2,N=3e4;
#define int long long 
 int n,a[N],tr[M];
 int c[N],d[N];
 
 int lowbit(int x){
     return x&-x;
 }
 void add(int x,int v){
     for(;x<M;x+=lowbit(x)) tr[x]+=v;
 }
 int q(int x){
     int t=0;
     for(;x;x-=lowbit(x)) t+=tr[x];
     return t;
 }
 void solve(){
     int i,j; 
     int ans=0;
     memset(tr,0,sizeof tr);
     for(i=1;i<=n;i++){
         c[i]=q(a[i]-1); add(a[i],1); 
     }
     
    memset(tr,0,sizeof tr);
    for(i=n;i>=1;i--){
          d[i]=q(a[i]-1); add(a[i],1);
     }
    
     for(i=1;i<=n;i++){
         ans+= c[i]*(n-i-d[i])+(i-c[i]-1)*d[i];
     }
    cout<<ans<<endl;
 }
 signed main(){
     //freopen("in","r",stdin);
    //freopen("out","w",stdout);
    int cas; cin>>cas;
     while(cas--){
     cin>>n;
     for(int i=1;i<=n;i++) cin>>a[i];
     solve();
     }
 }