1.二分查找
首先可以确定,矩形的面积不同,落在这些矩形中的概率是不同的,所以可以把矩形的面积当做矩形的权重
此处有一点需要注意,就是矩形面积的求法
题中说边缘的点也算在内,所以对于(x1,y1),(x2,y2),矩形的真实面积为(y2-y1)*(x2-x1)
为了使面积表示落在矩形内整点坐标的个数,我们让面积的值为(y2-y1+1)*(x2-x1+1)
累加矩形的面积和
设面积和为Sum,则rand()%Sum表示随机生成的0--sum-1之间的数,然后使用二分法查找生成的数应该在哪个矩形内
代码如下:
class Solution {
public:
vector<vector<int>>* rects;
vector<int> data{ 0 };
Solution(vector<vector<int>>& rects) {
this->rects = &rects;
for (vector<int>& vec : rects)
data.push_back(data.back() + (vec[3] - vec[1] + 1) * (vec[2] - vec[0] + 1));//data.size()=rects.size()+1
}
vector<int> pick() {
int r = rand() % data.back();
int start = 1;
int end = data.size() - 1;
while (start < end) {
int mid = start + (end - start >> 1);
if (data[mid] <= r)
start = mid + 1;
else
end = mid;
}
vector<int>& vec = (*rects)[start - 1];
return{ vec[0] + (r - data[start - 1]) % (vec[2] - vec[0] + 1), vec[1] + (r - data[start - 1]) / (vec[2] - vec[0] + 1) };
}
};