POJ 2002 Squares (已知正方形对角线两点推另外两点，STL set)

RuHua27

于 2014-07-23 22:38:01 发布

阅读量937

点赞数

CC 4.0 BY-SA版权

分类专栏： POJ 计算几何

本文链接：https://blog.csdn.net/u013112097/article/details/38071711

POJ 同时被 2 个专栏收录

37 篇文章

订阅专栏

计算几何

30 篇文章

订阅专栏

本文介绍了一种通过枚举正方形对角线上的两个点来检测一组点中是否存在正方形的算法。利用C++实现，该算法通过查找剩余两个顶点是否存在于预定义的点集合中来判断正方形的存在性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

枚举正方形对角线两点，推出另外两点，看在不在 set 里。（set 里 count() 返回整数，且只为 0 或 1 find() 返回迭代器）

#include <iostream>
#include <algorithm>
#include <utility>
#include <set>
using namespace std;

typedef pair<int, int> Point;
const int maxn = 1010;
int x[maxn], y[maxn];
set<Point> pset;

int main()
{
    int n;

    while (cin >> n && n) {
        pset.clear();
        for (int i = 0; i < n; i++) {
            cin >> x[i] >> y[i];
            pset.insert(make_pair(x[i], y[i]));
        }
        int cnt = 0;
        for (int i = 0; i < n; i++) {
            for (int j = i+1; j < n; j++) {
                int mx = x[i] + x[j], dx = max(x[i], x[j]) - min(x[i], x[j]);
                int my = y[i] + y[j], dy = max(y[i], y[j]) - min(y[i], y[j]);
                if ((mx+dy) & 1 || (my+dx) & 1) continue;
                int sg = ((x[i]-x[j]) * (y[i]-y[j]) < 0) ? 1 : -1;
                if (pset.count(make_pair((mx+dy)/2, (my+sg*dx)/2)) &&
                    pset.count(make_pair((mx-dy)/2, (my-sg*dx)/2)))
                    cnt++;
            }
        }
        cout << cnt / 2 << endl;
    }

    return 0;
}