Leedcode-构造矩形

自己写的

class Solution:
    def constructRectangle(self, area: int) -> List[int]:
        # 计算给定面积的平方根
        root = area ** 0.5

        # 初始化结果列表，默认为 [1, area]，即长为面积，宽为1的情况
        temp = [1, area]

        # 如果面积是一个完全平方数，则直接返回 [sqrt(area), sqrt(area)]
        if root.is_integer():
            return [int(root), int(root)]
        else:
            # 如果不是完全平方数，打印平方根的整数部分以供调试
            print(int(root))

            # 从2开始遍历到平方根的整数部分，寻找面积的因子
            for i in range(2, int(root) + 1):
                # 如果i是面积的因子
                if area % i == 0:
                    # 更新结果列表，确保长宽乘积等于给定面积
                    temp = [area // i, i]

        # 确保返回的结果列表中，第一个元素（长）大于或等于第二个元素（宽）
        if temp[0] < temp[1]:
            temp[0], temp[1] = temp[1], temp[0]

        return temp

 gpt优化：

from typing import List

class Solution:
    def constructRectangle(self, area: int) -> List[int]:
        # 计算给定面积的平方根，并将其转换为整数
        root = int(area ** 0.5)
        
        # 从平方根的整数部分向下遍历
        for i in range(root, 0, -1):
            # 如果i是面积的因子
            if area % i == 0:
                # 找到一个因子组合，直接返回
                return [area // i, i]
        
        # 理论上，代码不会运行到这里，因为1始终是任何正整数的因子
        return [area, 1]