【坚持每日一题7.26】497. 非重叠矩形中的随机点

给定一个非重叠轴对齐矩形的列表 rects,写一个函数 pick 随机均匀地选取矩形覆盖的空间中的整数点。

提示:

整数点是具有整数坐标的点。
矩形周边上的点包含在矩形覆盖的空间中。
第 i 个矩形 rects [i] = [x1,y1,x2,y2],其中 [x1,y1] 是左下角的整数坐标,[x2,y2] 是右上角的整数坐标。
每个矩形的长度和宽度不超过 2000。
1 <= rects.length <= 100
pick 以整数坐标数组 [p_x, p_y] 的形式返回一个点。
pick 最多被调用10000次。

示例 1:

输入:
["Solution","pick","pick","pick"]
[[[[1,1,5,5]]],[],[],[]]
输出:
[null,[4,1],[4,1],[3,3]]
示例 2:

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

输入语法的说明:

输入是两个列表:调用的子例程及其参数。Solution 的构造函数有一个参数,即矩形数组 rects。pick 没有参数。参数总是用列表包装的,即使没有也是如此。

java代码:

class Solution {

   int[][] rects;
   List psum = new ArrayList<>();
   int tot = 0;
   Random rand = new Random();

   public Solution(int[][] rects) {
       this.rects = rects;
       for (int[] x : rects){
           tot += (x[2] - x[0] + 1) * (x[3] - x[1] + 1);
           psum.add(tot);
       }
   }

   public int[] pick() {
       int targ = rand.nextInt(tot);

       int lo = 0;
       int hi = rects.length - 1;
       while (lo != hi) {
           int mid = (lo + hi) / 2;
           if (targ >= psum.get(mid)) lo = mid + 1;
           else hi = mid;
       }

       int[] x = rects[lo];
       int width = x[2] - x[0] + 1;
       int height = x[3] - x[1] + 1;
       int base = psum.get(lo) - width * height;
       return new int[]{x[0] + (targ - base) % width, x[1] + (targ - base) / width};
   }
}


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

你可能感兴趣的:(【坚持每日一题7.26】497. 非重叠矩形中的随机点)