Python Leetcode(492. 构造矩形)

Python Leetcode(492. 构造矩形)

作为一位web开发者， 懂得怎样去规划一个页面的尺寸是很重要的。 现给定一个具体的矩形页面面积，你的任务是设计一个长度为 L 和宽度为 W 且满足以下要求的矩形的页面。要求：

1. 你设计的矩形页面必须等于给定的目标面积。

2. 宽度 W 不应大于长度 L，换言之，要求 L >= W 。

3. 长度 L 和宽度 W 之间的差距应当尽可能小。

你需要按顺序输出你设计的页面的长度 L 和宽度 W。

示例：

输入: 4
输出: [2, 2]
解释: 目标面积是 4， 所有可能的构造方案有 [1,4], [2,2], [4,1]。
但是根据要求2，[1,4] 不符合要求; 根据要求3，[2,2] 比 [4,1] 更能符合要求. 所以输出长度 L 为 2， 宽度 W 为 2。

说明:

给定的面积不大于 10,000,000 且为正整数。
你设计的页面的长度和宽度必须都是正整数。

Solution:(刚开始思路有问题，想的是从矩形面积的一半开始遍历，实际上对于面积来说，恰好满足题意的最好情况是正方形，所以应该从面积的平方根开始遍历，向前遍历寻找较大的宽度即可。)

class Solution(object):
    def constructRectangle(self, area):
        """
        :type area: int
        :rtype: List[int]
        """
        import math
        width, val = 1, 0
        for width_val in range(1, int(math.sqrt(area))+1):
            if area % width_val == 0:
                width = max(width_val, val)
                val = width
        return [area/width, width]

这是最开始的思路，其实向前遍历更方便，找到就可以返回列表了。

class Solution(object):
    def constructRectangle(self, area):
        """
        :type area: int
        :rtype: List[int]
        """
        import math
        for width in range(int(math.sqrt(area)), 0, -1):
            if area % width == 0:
                return [area/width, width]