题意:一个数字序列,i,j,k,l四个下标关系为(i<j<k<l),问有多少组下标满足ai=ak,aj=al
n的范围是3000,显然暴力O(n4)是不行的,所以需要优化,我们需要解决的问题是在j之前有多少个ai=ak,有k之后有多少aj=al,我们开一个二维数组sum记录前多少个数有多少个某个数,ai=ak的个数为sum[j-1][ai],aj=al的个数为sum[n][aj]-sum[k][aj],n>=a,此情况下可以组成满足条件的四元组的个数为sum[j-1][ai]*(sum[n][aj]-sum[k][aj]),我们将所有的j,k列举求和即可
#include<map>
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<deque>
#include<cstring>
using namespace std;
deque<int> dqmin, dqmax;
const int N = 3030;
const int mod = 1e9;
long long arr[N],sum[N][N];
int main() {
ios::sync_with_stdio(false);
int t;
cin >> t;
while (t--) {
memset(sum, 0, sizeof(sum));
int n;
long long ans = 0;
cin >> n;
for (int i = 1; i <= n; i++) {
cin >> arr[i];
for (int j = 1; j <= n; j++) {
sum[i][j] = sum[i - 1][j];
}
sum[i][arr[i]]++;
}
for (int i = 2; i <= n - 2; i++) {
for (int k = i + 1; k <= n - 1; k++) {
ans = ans + sum[i - 1][arr[k]] * (sum[n][arr[i]] - sum[k][arr[i]]);
}
}
cout << ans << endl;
}
}