JAVA程序设计：非重叠矩形中的随机点（LeetCode：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 没有参数。参数总是用列表包装的，即使没有也是如此。

思路：这题有点坑啊，超时的代码提示说解答错误？？？导致我一直以为是自己哪里写错了，后来改成二分直接过了。。。

本题我的大致思路是对所有在矩形内点的数量取随机数，然后在相应矩形内找到这个点即可。

class Solution {

	private int len;
	private long size;
	private long[] nums;
	private long[] prev;
	private long[][] rects;
	
    public Solution(int[][] rects) {
    	this.rects=new long[rects.length][rects[0].length];
        for(int i=0;i<rects.length;i++)
        	for(int j=0;j<rects[0].length;j++)
        		this.rects[i][j]=(long)rects[i][j];
        this.len=rects.length;
        this.nums=new long[len];
        this.prev=new long[len];
        for(int i=0;i<len;i++) {
        	nums[i]=(long)((long)rects[i][2]-rects[i][0]+1)*((long)rects[i][3]-rects[i][1]+1);
        	size+=nums[i];
        	if(i>0) prev[i]=prev[i-1];
        	prev[i]+=nums[i];
        }
    }
    
    public int[] pick() {
        long index=(long)(Math.random()*size+1);
        int p=0;
        int l=0,r=len-1;
        while(l<=r) {
        	int mid=(l+r)/2;
        	if(prev[mid]<index) {
        		p=mid+1;
        		l=mid+1;
        	}
        	else
        		r=mid-1;
        }
        index=index-(p==0?0:prev[p-1]);
        long x=this.rects[p][2]-this.rects[p][0]+1;
        long y=this.rects[p][3]-this.rects[p][1]+1;
        long xx=(index/y),yy=(index%y);
        //System.out.println(xx+" "+yy);
        if(yy==0 && xx>0) xx--; 
        x=this.rects[p][0]+xx;
        y=this.rects[p][1]+yy;
        int[] res= {(int)x,(int)y};
        return res;
    }
}