Leetcode-593. 有效的正方形

链接

593. 有效的正方形

题目

给定2D空间中四个点的坐标 p1, p2, p3 和 p4，如果这四个点构成一个正方形，则返回 true 。

点的坐标 pi 表示为 [xi, yi] 。输入 不是 按任何顺序给出的。

一个 有效的正方形 有四条等边和四个等角(90度角)。

示例

示例 1:
输入: p1 = [0,0], p2 = [1,1], p3 = [1,0], p4 = [0,1]
输出: True

示例 2:
输入：p1 = [0,0], p2 = [1,1], p3 = [1,0], p4 = [0,12]
输出：false

示例 3:
输入：p1 = [1,0], p2 = [-1,0], p3 = [0,1], p4 = [0,-1]
输出：true

说明

• p1.length == p2.length == p3.length == p4.length == 2
• -10e4 <= xi, yi <= 10e4

思路

计算四条边和两条对角线的距离，这 6 条边是否符合：

• 4 条短边相等
• 2 条长边相等
• 长边是短边的 sqrt(2) 倍

C++ Code

class Solution {
public:
    inline int caculate(vector<int>& p1, vector<int>& p2){
        return (p1[0] - p2[0]) * (p1[0] - p2[0]) + (p1[1] - p2[1]) * (p1[1] - p2[1]);
    }
    bool validSquare(vector<int>& p1, vector<int>& p2, vector<int>& p3, vector<int>& p4) {
        unordered_map<int, int> map;
        map[caculate(p1, p2)]++;
        map[caculate(p1, p3)]++;
        map[caculate(p1, p4)]++;
        map[caculate(p2, p3)]++;
        map[caculate(p2, p4)]++;
        map[caculate(p3, p4)]++;
        for(auto it : map){
            if(it.second == 4 && map[it.first * 2] == 2){
                return true;
            }
        }
        return false;
    }
};