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]
 

提示：

1 <= rects.length <= 100
rects[i].length == 4
-109 <= ai < xi <= 109
-109 <= bi < yi <= 109
xi - ai <= 2000
yi - bi <= 2000
所有的矩形不重叠。
pick 最多被调用 104 次。

解题思路：（1）找出所有矩形包含的点的总个数s，遍历rects数组，对每一个rects[i]，其中点的个数是（rects[2]-rects[0]）*（rects[3]-rects[1]）。遍历的过程中同时记录前i个矩形共包含的点的个数。

（2）产生一个随机数r，范围是0~s-1

（3）找出随机数r指向的点在第几个矩阵中，二分查找法，找到第一个大于随机数的矩阵（r从0开始，相等的时候随机数r指向下一个矩阵的首个点）。

（4）找出r指向的点在该矩阵中的位置，设置点的次序按行排列。

class Solution {
    int s;
    int len;
    int[] si;
    int[][] rects;
    public Solution(int[][] rects) {
        this.rects = rects;
        s = 0;
        int xlen = 0;
        int ylen = 0;
        len = rects.length;
        si = new int[len];
        for(int i = 0; i < len; i++){
            xlen = rects[i][2] - rects[i][0] + 1;
            ylen = rects[i][3] - rects[i][1] + 1;
            s += xlen * ylen;
            si[i] = s;
        }
    }
    
    public int[] pick() {
        Random r = new Random();
        int ran = r.nextInt(s);
        int left = 0;
        int right = len - 1;
        int temp;
        while(left < right){
            int mid = (left + right) / 2;
            if(si[mid] <= ran)
                left = mid + 1;
            else
                right = mid;
        }
        if(left > 0)
            temp = ran - si[left - 1];
        else
            temp = ran;
        int retx = rects[left][0] + temp % (rects[left][2] - rects[left][0] + 1);
        int rety = rects[left][1] + temp / (rects[left][2] - rects[left][0] + 1);
        int[] ret = {retx, rety};
        return ret;
    }
}

/**
 * Your Solution object will be instantiated and called as such:
 * Solution obj = new Solution(rects);
 * int[] param_1 = obj.pick();
 */