LeetCode 593. 有效的正方形(数学)

1. 题目

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

一个点的坐标(x,y)由一个有两个整数的整数数组表示。

示例:
输入: p1 = [0,0], p2 = [1,1], p3 = [1,0], p4 = [0,1]
输出: True
 
注意:
所有输入整数都在 [-1000010000] 范围内。
一个有效的正方形有四个等长的正长和四个等角(90度角)。
输入点没有顺序。

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/valid-square
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

  • 4个点可以组成6条边,最长的两条是对角线
  • 对角线要垂直,且长度相等
class Solution {	//C++
public:
    bool validSquare(vector<int>& p1, vector<int>& p2, vector<int>& p3, vector<int>& p4) {
        set<vector<int>> s;
    	vector<vector<int>> p = {p1,p2,p3,p4};
    	vector<vector<int>> vec;
        for(auto& pi : p)
            s.insert(pi);
        if(s.size() < 4) return false;//有重合点
    	int dx, dy, dis;
    	for(int i = 0, j; i < 3; ++i)
    		for(j = i+1; j < 4; ++j)
    		{
    			dx = p[i][0]-p[j][0];
    			dy = p[i][1]-p[j][1];
    			vec.push_back({dx,dy});
    		}
		sort(vec.begin(), vec.end(),[&](auto a, auto b){
			return a[0]*a[0]+a[1]*a[1] < b[0]*b[0]+b[1]*b[1];
		});
		//最长的两条边
		int dis1 = vec[4][0]*vec[4][0]+vec[4][1]*vec[4][1];
		int dis2 = vec[5][0]*vec[5][0]+vec[5][1]*vec[5][1];
		//两向量 模长相等,且垂直
		return (dis1==dis2) && (vec[4][0]*vec[5][0]+vec[4][1]*vec[5][1]==0);
    }	
};

12 ms 26.6 MB

class Solution:# py3
    def validSquare(self, p1: List[int], p2: List[int], p3: List[int], p4: List[int]) -> bool:
        p = [p1,p2,p3,p4]
        vec = []
        s = set(tuple(pi) for pi in p)
        if len(s) < 4:
            return False
        for i in range(3):
            for j in range(i+1, 4):
                dx = p[i][0]-p[j][0]
                dy = p[i][1]-p[j][1]
                vec.append([dx,dy])
        vec = sorted(vec, key=lambda a:a[0]*a[0]+a[1]*a[1])
        dis1 = vec[4][0]*vec[4][0]+vec[4][1]*vec[4][1];
        dis2 = vec[5][0]*vec[5][0]+vec[5][1]*vec[5][1];
        return (dis1==dis2) and (vec[4][0]*vec[5][0]+vec[4][1]*vec[5][1]==0)

44 ms 13.6 MB

猜你喜欢

转载自blog.csdn.net/qq_21201267/article/details/106926403