[LC593]有效的正方形-Valid Square

题目描述

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

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

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

链接：https://leetcode.cn/problems/valid-square

基本思路

正方形的一个集合特点是，任意三个点都可以组成一个等腰直角三角形，等腰直角三角形是易于判断的，那么我们可以按顺序的选择四个点中的三个，依次去检测是否是等腰三角形，即可确认最后四个点是否可以组成正方形。

代码

Golang

func validSquare(p1 []int, p2 []int, p3 []int, p4 []int) bool {
	ok, len := isTriangle(p1, p2, p3, -1)
	if !ok {
		return false
	}
	ok, len = isTriangle(p4, p2, p3, len)
	if !ok {
		return false
	}
	ok, len = isTriangle(p1, p4, p3, len)
	if !ok {
		return false
	}
	ok, _ = isTriangle(p1, p2, p4, len)
	return ok
}

func min(i1 int, i2 int) int {
	if i1 > i2 {
		return i2
	}
	return i1
}

func getDist(p1 []int, p2 []int) int {
	return (p1[0]-p2[0])*(p1[0]-p2[0]) + (p1[1]-p2[1])*(p1[1]-p2[1])
}

func isTriangle(p1 []int, p2 []int, p3 []int, len int) (bool, int) {
	l1 := getDist(p1, p2)
	l2 := getDist(p2, p3)
	l3 := getDist(p1, p3)
	ok := l1+l2 == l3 || l1+l3 == l2 || l2+l3 == l1
	if !ok {
		return false, -1
	}
	if len == -1 {
		len = min(l1, l2)
	} else if len == 0 || len != min(l1, l2) {
		return false, -1
	}
	return ok, len
}