https://codeforces.com/contest/1400/problem/D
感觉这题是个div3A难度的题啊怎么放在D的位置大家都5分钟秒了啊。。。
枚举a[j],a[k],那么他们对答案的贡献就是(1,j-1)中a[k]的数量乘以(k+1,n)中a[j]的数量,用个前缀后缀和数组维护下就好了
#include<bits/stdc++.h>
#define pb push_back
using namespace std;
typedef long long ll;
const int maxl=3010;
int n,m,cas,k,cnt,tot;ll ans;
int a[maxl],b[maxl];
ll num[maxl],suf[maxl][maxl],pre[maxl][maxl];
char s[maxl];
bool in[maxl];
inline void prework()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
num[i]=0,scanf("%d",&a[i]);
for(int j=1;j<=n;j++)
suf[n+1][j]=0;
for(int i=n;i>=1;i--)
{
for(int j=1;j<=n;j++)
suf[i][j]=suf[i+1][j];
suf[i][a[i]]++;
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
pre[i][j]=pre[i-1][j];
pre[i][a[i]]++;
}
}
inline void mainwork()
{
ans=0;
for(int i=2;i<=n;i++)
for(int j=i+1;j<n;j++)
ans+=pre[i-1][a[j]]*suf[j+1][a[i]];
}
inline void print()
{
printf("%lld\n",ans);
}
int main()
{
int t=1;
scanf("%d",&t);
for(cas=1;cas<=t;cas++)
{
prework();
mainwork();
print();
}
return 0;
}