力扣算法_497 非重叠矩形中的随机点

题目

给定一个由非重叠的轴对齐矩形的数组 rects ，其中 rects[i] = [ai, bi, xi, yi] 表示 (ai, bi) 是第 i 个矩形的左下角点，(xi, yi) 是第 i 个矩形的右上角点。设计一个算法来随机挑选一个被某一矩形覆盖的整数点。矩形周长上的点也算做是被矩形覆盖。所有满足要求的点必须等概率被返回。

在给定的矩形覆盖的空间内的任何整数点都有可能被返回。

请注意 ，整数点是具有整数坐标的点。

实现 Solution 类:

• Solution(int[][] rects) 用给定的矩形数组 rects 初始化对象。
• int[] pick() 返回一个随机的整数点 [u, v] 在给定的矩形所覆盖的空间内。

示例

示例1:

输入: 
["Solution", "pick", "pick", "pick", "pick", "pick"]
[[[[-2, -2, 1, 1], [2, 2, 4, 6]]], [], [], [], [], []]
输出: 
[null, [1, -2], [1, -1], [-1, -2], [-2, -2], [0, 0]]

解释：
Solution solution = new Solution([[-2, -2, 1, 1], [2, 2, 4, 6]]);
solution.pick(); // 返回 [1, -2]
solution.pick(); // 返回 [1, -1]
solution.pick(); // 返回 [-1, -2]
solution.pick(); // 返回 [-2, -2]
solution.pick(); // 返回 [0, 0]

思路

方法一：先随机选择一个长方形，然后再随机选择该长方形内的点；该方法解答错误，因为无法做到等概率随机取任意一点。

方法二：先计算这些长方形总共包含多少点，然后从这些点中随机选择一个；但如果将所有点都一一罗列会时间超出限制，因此，可以利用数组存储前n个长方形总共包含的点数，然后随机从总数中选择一个数，判断其属于哪个长方形。

代码

方法一：

class Solution:

    def __init__(self, rects: List[List[int]]):
        self.rects = rects
        self.n = len(rects)
        return

    def pick(self) -> List[int]:
        i = random.randint(0, self.n - 1)
        return [random.randint(self.rects[i][0], self.rects[i][2]), random.randint(self.rects[i][1], self.rects[i][3])]

方法二：

class Solution:

    def __init__(self, rects: List[List[int]]):
        self.rects = rects
        self.sum = [0]
        for a, b, x, y in rects:
            self.sum.append(self.sum[-1] + (x - a + 1) * (y - b + 1))

    def pick(self) -> List[int]:
        k = random.randrange(self.sum[-1])
        rectIndex = 0
        for i, nn in enumerate(self.sum):
            if nn > k:
                rectIndex = i
                break
        a, b, _, y = self.rects[rectIndex-1]
        da, db = divmod(k - self.sum[rectIndex-1], y - b + 1)
        return [a + da, b + db]