Zigzags

题意：一个数字序列，i,j,k,l四个下标关系为(i<j<k<l),问有多少组下标满足ai=ak,aj=al

n的范围是3000，显然暴力O(n⁴)是不行的，所以需要优化，我们需要解决的问题是在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;
	} 
}