暑假集训Day12 G (定位计数)

题目链接在这里：Problem - G - Codeforces

题目大意是必须要找一对相等的数，然后这两对数的位置是要交叉的。

最朴素的暴力可以是枚举一对数，然后看这对数中间和前后有多少相同的对。

关于区间相同数的个数我们可以通过维护一个前缀和最后用O(1)的复杂度跑出来。

由于我们要求的四个数里面只有两个不同的数，所以优化一下只要枚举这两个数的位置就好了，剩下的可以用前缀和解决

#include "bits/stdc++.h"
using namespace std;
typedef long long LL;
const int MAX=3005;
LL t,n,a[MAX],tot;
LL cnt[MAX][MAX],cc[MAX];
struct Node{
    LL b[MAX],cn,no;
}stu[MAX];
bool cmp(Node x,Node y){
    return x.cn<y.cn;
}
LL mul(LL x){return ((x==1||x==0)?1:x*mul(x-1));}
LL C(LL x,LL y){
    return mul(x)/(mul(y)*mul(x-y));
}
int main(){
    freopen ("g.in","r",stdin);
    freopen ("g.out","w",stdout);
    LL i,j,ans;
    scanf("%d",&t);
    while (t--){
        scanf("%d",&n);
        memset(cnt,0,sizeof(cnt));
        tot=0;
        for (i=1;i<=n;i++){
            for (j=1;j<=n;j++) cnt[i][j]+=cnt[i-1][j];
            scanf("%d",a+i),cnt[i][a[i]]++;
            cc[a[i]]++;
        }
        for (i=1;i<=n;i++)
            if (cc[i]>1){
                stu[++tot].cn=cc[i];
                stu[tot].no=i;
            }
        sort(stu+1,stu+tot+1,cmp);
        i=1;ans=0;
        while (stu[i].cn<4 && i<=tot) i++;
        for (;i<=tot;i++) ans+=C(stu[i].cn,4);
        ans=0;//cout<<i<<endl;
        for (i=1;i<n;i++)
            for (j=i+1;j<=n;j++){
                ans+=(cnt[n][a[i]]-cnt[j][a[i]])*cnt[i-1][a[j]];
                //cout<<i<<' '<<j<<(cnt[n][a[i]]-cnt[j][a[i]])*cnt[i-1][a[j]]<<endl;
            }
        printf("%lld
",ans);
    }
    return 0;
}