【第13届景驰-埃森哲杯广东工业大学ACM程序设计大赛】E题 回旋星空

题目链接

https://www.nowcoder.com/acm/contest/90/E

题目

链接：https://www.nowcoder.com/acm/contest/90/E
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 131072K，其他语言262144K
64bit IO Format: %lld
题目描述
曾经有两个来自吉尔尼斯的人(A和C)恋爱了,他们晚上经常在一起看头上的那片名为假的回旋星空,

有一天他们分手了,A想通过回旋星空测量他们之间的复合指数,测量的规则是,

计算回旋图标的个数,即选中三颗星星,分别作为回旋图标的起点,拐点和终点,假设现在有三个

星星分别为i,j,k,如果d(a[i],a[j]) == d(a[j],a[k])则表示找到了一个回旋图标,其中d(x,y)表示这两个点的欧氏距离

为了给它很大的希望(i,j,k)和(k,j,i)被认为是两个不同的回旋图标

A花了一晚上终于把整片星空映射到了一张二平面图上,由于星星太多以至于A有点懵逼,所以

你能帮帮他吗,要不然他可能真的WA的一声哭了出来
作为埃森哲公司的一员，你在解决问题的同时也向A介绍了埃森哲公司的业务范围。
为了全方位地满足客户的需求，正在不断拓展自身的业务服务网络，包括管理及信息技术咨询、企业经营外包、企业联盟和风险投资。除了以产品制造业、通信和高科技、金融服务、资源、政府机构等不同行业划分服务内容之外，还从以下几方面提供咨询服务：
1.客户关系管理
2.业务解决方案
3.电子商务
4.供应链管理

输入描述:
第一行一个整数T(T<=10),表示组数

对于每组数据有一个n,表示有n个小星星(0< n < 1000)

接下来跟着n行,每行跟两个整数xi和yi表示每个星星的坐标(-10000< xi, yi<10000)

输出描述:
对于每组数据,如果没有找到回旋图标输出”WA”,否则输出找到图标个数
示例1
输入
2
2
1 0
0 1
3
1 0
0 1
0 0
输出
WA
2
备注:
没有重复的星星，且选中的三个星星是互相不一样的（即下标不同）
欧氏距离即直线距离

分析

对于任意一个顶点i，求出i与其他n-1个顶点所连边的距离，距离相等的边放在一个集合。求出有cnt个集合，每个集合的大小分别是x=a[i]，则集合i的x个边能组成C（x，2）*2个回旋图标，C为求组合数。枚举n个顶点，累加各顶点的回旋图标数即为所求。

代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <map>
#include <vector>

using namespace std;
const int maxn=1e3+100;
typedef long long ll;
vector<int> g[maxn];
map<int,int> m;
double dis[maxn][maxn];
struct point
{
    int x,y,cnt;
    int a[maxn];
}p[maxn];

double Dis(int i,int j)
{
    return (p[i].x-p[j].x)*(p[i].x-p[j].x)
    +(p[i].y-p[j].y)*(p[i].y-p[j].y);
}

void getDis(int n)
{
    for(int i=1;i<=n;i++)
    {
        for(int j=i+1;j<=n;j++)
        {
            dis[i][j]=dis[j][i]=Dis(i,j);
        }
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        memset(p,0,sizeof(p));
        for(int i=0;i<=n;i++) g[i].clear();
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&p[i].x,&p[i].y);
        }

        getDis(n);
        ll ans=0;
        for(int i=1;i<=n;i++)
        {
            p[i].cnt=0;
            int &k=p[i].cnt;
            m.clear();
            for(int j=1;j<=n;j++)if(i!=j)
            {
                int tmp=dis[i][j];
                if(m[tmp])
                {
                    p[i].a[m[tmp]]++;
                }
                else
                {
                    k++;
                    m[tmp]=k;
                    p[i].a[m[tmp]]++;
                }
            }
            for(int j=1;j<=k;j++)
            {
                ll sum=p[i].a[j];
                ans+=sum*(sum-1);
            }
        }
        if(ans)
            printf("%lld\n",ans);
        else
            printf("WA\n");
    }
    return 0;
}