题目描述
曾经有两个来自吉尔尼斯的人(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行,每行跟两个整数xi和yi表示每个星星的坐标(-10000< xi, yi<10000)
输出描述:
对于每组数据,如果没有找到回旋图标输出”WA”,否则输出找到图标个数
示例1
输入
2
2
1 0
0 1
3
1 0
0 1
0 0
输出
WA
2
对每个点,计算其他n-1个点到他的距离再排序后,到该点距离相同的点就排到一起了,假如一共有t个,那么以该点为中心的回旋就有A2tAt2个,也就是t*(t-1)个。把每个点的回旋数加起来就是答案
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=1005;
int x[MAXN],y[MAXN];
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n;
scanf("%d",&n);
for(int i=0;i<n;i++)
scanf("%d%d",&x[i],&y[i]);
ll res=0;
for(int i=0;i<n;i++)
{
vector<int> dis;
for(int j=0;j<n;j++)//计算每一点与第i个点的距离 ,由于dis是在循环内声明的,所以每次循环dis只保存其余n点到第i点的距离
if(j!=i)dis.push_back((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
sort(dis.begin(),dis.end());//排序后距离相同的数字在一起,比如说11122344566
for(int t = 0;t < dis.size();)//对于 111,根据排列组合,可以算出有3*2=6种排列
{
int count = 1;
int tt = 0;//为了控制t挪动的位置,才引入tt,使t每次移动到下一个不相同的数字
while(dis[t] == dis[t + 1] && (t + 1) < dis.size()) {
count++;
t++;
tt--;
}
t += count + tt;
res += count * (count - 1);
}
}
if(res==0)printf("WA\n");
else printf("%lld\n",res);
}
return 0;
}