第13届景驰-埃森哲杯广东工业大学ACM程序设计大赛

E：回旋星空

链接：https://www.nowcoder.com/acm/contest/90/E

题目描述

曾经有两个来自吉尔尼斯的人(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的一声哭了出来

输入描述:

第一行一个整数T(T<=10),表示组数

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

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

输出描述:

对于每组数据,如果没有找到回旋图标输出”WA”,否则输出找到图标个数

枚举点k，将它和所有点的距离存下来，排序，查看每个距离相同的有多少个，若大于等于2（假设为m个）,当前距离产生的回旋图标的个数为m*(m-1)。

#include<bits/stdc++.h>
using namespace std;
#define N 1340
#define EPS 1e-5
set <int> mp;
struct node
{
    int x, y;
}point[N];

int distant(int x1, int y1, int x2, int y2)
{
    return (x1-x2)*(x1-x2) + (y1-y2)*(y1-y2);
}
int main()
{
    int T;
    long long sum = 0, num;
    scanf("%d", &T);
    while(T--) {
        mp.clear();
        int n;
        sum = 0;
        scanf("%d", &n);
        for(int i = 1; i <= n; ++i)
            scanf("%d%d", &point[i].x, &point[i].y);
        for(int i = 1; i <= n; ++i) {   //枚举k
            vector <int> dis;
            for(int j = 1; j <= n; ++j) {
                if(i == j)
                    continue;
                dis.push_back(distant(point[i].x,point[i].y,point[j].x, point[j].y));
            }
            sort(dis.begin(), dis.end());
            num = 0;
            for (int j = 0, r = 0; j < dis.size(); j = r) {
                while(r<dis.size() && dis[j]==dis[r])   r++;
                sum += (r-j)*(r-j-1);
            }
        }
        if(sum == 0)
            printf("WA\n");
        else
            printf("%lld\n", sum);;
    }
    return 0;
}