[LeetCode] 593. 有效的正方形

本文介绍了一种算法,用于判断平面直角坐标系中任意四点是否能构成正方形。该算法通过计算各点间距离,判断是否有两种固定长度(边长与对角线),并确保所有边长相等且所有对角线相等。

传送门

题目描述

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

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

示例:

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

注意:
1. 所有输入整数都在 [-10000,10000] 范围内。
2. 一个有效的正方形有四个等长的正长和四个等角(90度角)。
3. 输入点没有顺序。 注意:这个是关键条件

我的思路

这道题乍一看感觉是很简单的一道题,但是一定要注意到关键条件,因为输入的 4 个点是没有顺序的,所以我们不知道,任意两点之间的连线是正方形的边还是对角线,这给题目增加了困难

我的想法是,逐个计算点与点之间的距离,设置两个变量 edge 和 diagonal 分别用来记录正方形边长和对角线长,正方形中,边长和对角线分别都是等长的,所以计算任意两点之间的距离,应该只会有两种可能(即:一个边长,一个对角线长)
那么,我们在计算的过程中,如果出现了第三种边长,则直接返回 false

注:因为题中没有特殊要求,所以我们甚至不需要分辨哪个是边哪个是对角线

AC代码

class Solution {
    public boolean validSquare(int[] p1, int[] p2, int[] p3, int[] p4) {
        if (p1 == null || p2 == null || p3 == null || p4 == null)
            return false;
        if (p1.length != 2 || p2.length != 2 || p3.length != 2 || p4.length != 2)
            return false;
        int[][] points = {p1, p2, p3, p4};
        double edge = 0;
        double diagonal = 0;
        for (int i = 0; i < points.length; i++){
            for (int j = i + 1; j < points.length; j++){
                double length = lengthOf(points[i], points[j]);
                if (length == 0) return false;
                if (edge == 0){
                    edge = length;
                }
                if (diagonal == 0 && length != edge){
                    diagonal = length;
                }
                if (length != edge && length != diagonal){
                    return false;
                }
            }
        }
        return true;
    }

    private static double lengthOf(int[] a, int[] b){
        return Math.sqrt(Math.pow(a[0] - b[0], 2) + Math.pow(a[1] - b[1], 2));
    }
}

AC代码

class Solution {
    public boolean validSquare(int[] p1, int[] p2, int[] p3, int[] p4) {
        if (p1 == null || p2 == null || p3 == null || p4 == null)
            return false;
        if (p1.length != 2 || p2.length != 2 || p3.length != 2 || p4.length != 2)
            return false;
        int[][] points = {p1, p2, p3, p4};
        double edge = 0;
        double diagonal = 0;
        for (int i = 0; i < points.length; i++){
            for (int j = i + 1; j < points.length; j++){
                double length = lengthOf(points[i], points[j]);
                if (length == 0) return false;
                if (edge == 0){
                    edge = length;
                }
                if (diagonal == 0 && length != edge){
                    diagonal = length;
                }
                if (length != edge && length != diagonal){
                    return false;
                }
            }
        }
        return true;
    }

    private static double lengthOf(int[] a, int[] b){
        return Math.sqrt(Math.pow(a[0] - b[0], 2) + Math.pow(a[1] - b[1], 2));
    }
}
在解决 LeetCode 中的“木棍拼正方形”问题时,核心是通过深度优先搜索(DFS)与剪枝优化来判断是否可以将给定的木棍分成四组,每组的和相等。该问题可以归类为经典的“划分问题”或“分割子集问题”,通常与回溯算法结合使用。 ### 问题描述 给定一个整数数组 `matchsticks` 和一个整数 `total`,表示所有木棍的长度和。目标是判断是否可以将这些木棍拼成一个正方形,且每条边的长度相等。正方形有四条边,因此需要将木棍分成四组,每组的总长度为 `total / 4`。 ### 解题思路 1. **初步判断**: - 如果木棍总长度不能被 4 整除,则直接返回 `False`。 - 如果最长木棍的长度超过单边目标长度,则无法拼接正方形,直接返回 `False`。 2. **排序优化**: - 将木棍长度从大到小排序,优先尝试较长的木棍,有助于减少无效递归路径。 3. **深度优先搜索(DFS)**: - 使用 DFS 尝试将每根木棍放入当前未装满的边中。 - 如果当前边的长度已超过目标值,则剪枝。 - 如果某根木棍成功填满某条边,则优先处理下一条边。 4. **剪枝优化**: - 如果当前木棍与上一根木棍长度相同,且上一根木棍未被使用,则跳过当前木棍的尝试,避免重复计算。 - 当某条边首次尝试失败后,可以直接回溯,减少递归次数。 ### 代码实现 ```python def makesquare(matchsticks): total = sum(matchsticks) if total % 4 != 0: return False target = total // 4 matchsticks.sort(reverse=True) def dfs(index, sides): if index == len(matchsticks): return all(side == target for side in sides) for i in range(4): if sides[i] + matchsticks[index] <= target: if i > 0 and sides[i] == sides[i - 1]: continue # 避免重复尝试相同长度的边 sides[i] += matchsticks[index] if dfs(index + 1, sides): return True sides[i] -= matchsticks[index] if sides[i] == 0: break # 如果当前木棍无法放入任何边,直接剪枝 return False return dfs(0, [0] * 4) ``` ### 算法复杂度分析 - **时间复杂度**:最坏情况下为 $O(4^n)$,其中 $n$ 是木棍的数量。由于使用了剪枝优化,实际运行时间远低于理论上限。 - **空间复杂度**:$O(n)$,主要为递归栈的空间。 ### 示例 假设输入为 `matchsticks = [1,1,2,2,2]`,总长度为 8,每条边的目标长度为 2。算法会尝试将木棍分配到四条边中,并最终返回 `True`。 ### 优化建议 - **记忆化搜索**:可以引入缓存机制,避免重复计算相同状态。 - **位掩码优化**:使用位运算记录已使用的木棍,减少空间占用。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值