CF1400D Zigzags（思维）

Practice link： https://codeforces.ml/problemset/problem/1400/D

题意：给你一个数组，求（i、j、k、l) 满足

         的组合数量。

          数据范围： 4<=n<=3000   1<=a[ i ]<=n 

思路：数据范围很小，因此可以枚举 j 和 k ,只要求出位置 j 前面与 a[ k ] 相等的数目 x ，和位置 k 后面与 a[ j ]相等的数目 y ，最后答案加上x和y的乘积即可。

         用sum[ i ][ j ]来表示第 i 个数前面值为 j 的数的个数。在每次枚举中答案就是 ans=ans+sum[ j-1 ][ a[ k ] ] * (sum[ n ][ a[ j ] ] - sum[ k ][ a[ j ] ]) 。

代码：

//#include<bits/stdc++.h>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
#define ll long long
#define MOD 1000000007
#define INF 0x3f3f3f3f
#define mem(a,x) memset(a,x,sizeof(a))
#define lowbit(x) x&(-x)
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
 
int T;
int a[3005];
int sum[3005][3005];
int main()
{
    int n;
    cin>>T;
    while(T--){
        scanf("%d",&n);
        mem(sum,0);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            for(int j=1;j<=i;j++){
                sum[i][a[j]]++;
            }
        }
        ll ans=0;
        for(int j=2;j<=n-2;j++){
            for(int k=j+1;k<=n-1;k++){
                ans=ans+1ll*sum[j-1][a[k]]*(sum[n][a[j]]-sum[k][a[j]]);
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}