Zigzags

最新推荐文章于 2021-11-06 14:37:37 发布

原创最新推荐文章于 2021-11-06 14:37:37 发布 · 348 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

补题专栏收录该内容

42 篇文章

订阅专栏

本文探讨了在数字序列中寻找特定四元组(i,j,k,l)的高效算法，其中ai=ak且aj=al，i<j<k<l。通过使用二维数组sum来记录前缀中每个数值出现的次数，将原本O(n^4)的暴力解法优化至更高效的时间复杂度。适用于n≤3000的数据规模。

题意：一个数字序列，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;
	} 
}