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 ;
}
上一篇:CentOS7安装MySQL、Redis、RabbitMQ


下一篇:使用一年ESB感受