python 输入正方形_593. 有效的正方形（Python）

题目

难度：★★★☆☆

类型：平面几何

方法：数学

力扣链接请移步本题传送门

更多力扣中等题的解决方案请移步力扣中等题目录

给定二维空间中四点的坐标，返回四点是否可以构造一个正方形。

一个点的坐标（x，y）由一个有两个整数的整数数组表示。

示例:

输入: p1 = [0,0], p2 = [1,1], p3 = [1,0], p4 = [0,1]

输出: True

解答

这道题是初中数学题，正方形是菱形也是矩形，需要满足两个条件：

所有边长都相等（菱形）；

对角线相等（矩形）；

这里需要注意的是，我们并不知道哪两个点确定一条边，因此需要枚举计算欧氏距离，一共六种选择。本例中选用欧氏距离的平方进行比较。

留意特殊情况的处理。

from collections import Counter

class Solution:

def validSquare(self, p1, p2, p3, p4):

if len({tuple(p) for p in [p1, p2, p3, p4]}) != 4:

return False

def dist_square(p, q):

return (p[0] - q[0]) ** 2 + (p[1] - q[1]) ** 2

dist_list = [dist_square(p, q) for p, q in [(p1, p2), (p1, p3), (p1, p4), (p2, p3), (p2, p4), (p3, p4)]]

dist_counter = Counter(dist_list)

if len(dist_counter) != 2:

return False

if list(sorted(dist_counter.values())) != [2, 4]:

return False

dist_counter = {v: k for k, v in dist_counter.items()}

edge, diagonal = dist_counter[4], dist_counter[2]

if abs(edge * 2 - diagonal) > 1e-3:

return False

return True

如有疑问或建议，欢迎评论区留言~

有关更多力扣中等题的python解决方案，请移步力扣中等题解析