bzoj 4927: 第一题

Description

给定n根直的木棍，要从中选出6根木棍，满足：能用这6根木棍拼

出一个正方形。注意木棍不能弯折。问方案数。

正方形：四条边都相等、四个角都是直角的四边形。

Input

第一行一个整数n。

第二行包含n个整数ai，代表每根木棍的长度。

n ≤ 5000, 1 ≤ ai ≤ 10^7

Output

一行一个整数，代表方案数。

考虑正方形每条边由几条木棍组成，只有1122和1113两种情况，分别枚举统计一下

#include<bits/stdc++.h>
typedef long long i64;
int n,a[],ap=,as[],at[],t[];
i64 ans=,c3[];
int main(){
    scanf("%d",&n);
    for(int i=;i<=n;++i)scanf("%d",a+i);
    std::sort(a+,a+n+);
    for(int i=,j=;i<=n;i=j){
        int x=a[i];
        for(;j<=n&&a[j]==x;++j);
        c3[ap]=i64(j-i)*(j-i-)*(j-i-)/;
        as[ap]=x;
        at[ap++]=j-i;
    }
    for(int i=;i<=n;++i){
        for(int j=;j<ap;++j)if(as[j]>a[i]){
            ans+=c3[j]*t[as[j]-a[i]];
        }
        for(int j=;j<i&&a[i]+a[j]<=;++j)++t[a[i]+a[j]];
    }
    for(int i=,l,r;i<ap;++i)if(at[i]>=){
        i64 s=,s0=,s1;
        for(r=i-,l=;r>=l;--r){
            for(;r>=l&&as[l]+as[r]<as[i];++l);
            if(r<l||as[l]+as[r]!=as[i])continue;
            int x=at[l];
            if(r==l){
                s1=x*(x-)/;
                s+=s1*(x-)*(x-)/;
            }else{
                int y=at[r];
                s1=x*y;
                s+=s1*(x-)*(y-)/;
            }
            s+=s0*s1;
            s0+=s1;
        }
        ans+=at[i]*(at[i]-)/*s;
    }
    printf("%lld\n",ans);
    return ;
}