Leetcode 478. 在圆内随机生成点 (随机数生成题目)
程序员文章站
2022-05-22 13:08:50
...
这里我们仅利用C语言中的rand()函数,生成任意的一个随机整数
如何根据这个rand()函数生成[-1,1]之间的随机浮点数,答案是
a = 2*(double)rand()/RAND_MAX-1;
这里可以用平移伸缩变换的方法,我们只需要生成在原点,半径为1的随机点,然后通过平移以及伸缩变换就能得到想要的结果。
class Solution {
public:
double x, y, r;
Solution(double radius, double x_center, double y_center) {
x = x_center;
y = y_center;
r = radius;
}
vector<double> randPoint() {
double a, b;
do{
a = 2*(double)rand()/RAND_MAX-1;
b = 2*(double)rand()/RAND_MAX-1;
}while(a*a+b*b>1);
return {x+r*a,y+r*b};
}
};
方法二:计算分布函数
使用极坐标来看就很清晰,角度theta肯定是0到2pi均匀分布没问题,半径满足半径的平方均匀分布,所以写下如下代码
import random
import math
class Solution(object):
def __init__(self, radius, x_center, y_center):
"""
:type radius: float
:type x_center: float
:type y_center: float
"""
self.radius = radius
self.x = x_center
self.y = y_center
def randPoint(self):
"""
:rtype: List[float]
"""
r = (random.random() ** 0.5) * self.radius
#即 (random.random() * self.radius * self.radius) ** 0.5
theta = random.uniform(0, 2 * math.pi)
return [r * math.cos(theta) + self.x, r * math.sin(theta) + self.y]