问题描述
很久很久以前,有一个长度为n的数列A,数列中的每个数都不小于1且不大于n,且数列中不存在两个相同的数. 请统计有多少四元组(a,b,c,d)满足: 1. 1≤a<b<c<d≤n 2. Aa<Ab 3. Ac<Ad
输入说明
第一行输入一个整数T,表示有T组测试数据.
每组测试数据第一行包含一个整数n,第二行包含n个整数.
数据范围
1 <= T <= 100
1 <= n <= 50000
1 <= Ai <= n
输出说明
对于每组测试数据,输出一个整数表示四元组的数量.
输入样例
1 5 1 3 2 4 5
输出样例
4
代码如下:
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <string>
#include <algorithm>
#include <vector>
#include <deque>
#include <list>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <numeric>
#include <iomanip>
#include <bitset>
#include <sstream>
#include <fstream>
#include <limits.h>
#include <ctime>
#define debug "output for debug\n"
#define pi (acos(-1.0))
#define eps (1e-6)
#define inf (1<<28)
#define sqr(x) (x) * (x)
#define mod 1000000007
using namespace std;
typedef long long ll;
typedef unsigned long long ULL;
#define MAX 50005
ll n,a[MAX],f1[MAX],f2[MAX],c[MAX];
//f1[i]第i个数之前有多少个比它小的数,f2[i]第i个数之后有多少个比它大的数
ll lowbit(ll x)
{
return x&(-x);
}
void update(ll i,ll x)//在第i个位置上增加x
{
while(i<=n)
{
c[i]+=x;
i+=lowbit(i);
}
}
ll sum(ll i) //c[1-i]之间的和
{
ll s=0;
while(i>0)
{
s+=c[i];
i-=lowbit(i);
}
return s;
}
int main()
{
ll i,t;
scanf("%I64d",&t);
while(t--)
{
scanf("%I64d",&n);
for(i=1;i<=n;i++)
scanf("%I64d",&a[i]);
memset(c,0,sizeof(c));
for(i=1;i<=n;i++)
{
f1[i]=sum(a[i]);//统计a[i]之前有多少个比它小的数
update(a[i],1);//在a[i]位置上加1,表示已经存在
}
memset(c,0,sizeof(c));
for(i=n;i>=1;i--)
{
f2[i]=(n-i)-sum(a[i]);//倒着输入,第i个数后面有n-i个数,再看看这n-i个数中是不是存在比a[i]小的(即sum(a[i])),减去它们,就是a[i]后面所有比它大的
update(a[i],1);
}
ll ans=0,cnt=0;
for(i=1;i<=n-1;i++)//枚举c的位置
{
ans+=cnt*f2[i];//cnt表示输入顺序中第i-1个数之前有多少个数比第i-1个数小,在本题中也就是比b小的个数,i是c
cnt+=f1[i];
}
printf("%I64d\n",ans);
}
return 0;
}