题目:
给你一个点数组 points 和一个表示角度的整数 angle ,你的位置是 location ,其中 location = [posx, posy] 且 points[i] = [xi, yi] 都表示 X-Y 平面上的整数坐标。
最开始,你面向东方进行观测。你 不能 进行移动改变位置,但可以通过 自转 调整观测角度。换句话说,posx 和 posy 不能改变。你的视野范围的角度用 angle 表示, 这决定了你观测任意方向时可以多宽。设 d 为你逆时针自转旋转的度数,那么你的视野就是角度范围 [d - angle/2, d + angle/2] 所指示的那片区域。
对于每个点,如果由该点、你的位置以及从你的位置直接向东的方向形成的角度 位于你的视野中 ,那么你就可以看到它。
同一个坐标上可以有多个点。你所在的位置也可能存在一些点,但不管你的怎么旋转,总是可以看到这些点。同时,点不会阻碍你看到其他点。
返回你能看到的点的最大数目。
解答:
class Solution:
def visiblePoints(self, points: List[List[int]], angle: int, location: List[int]) -> int:
self.base = 0
def helper(point):
dx, dy = point[0] - location[0], point[1] - location[1]
if not dx and not dy:
self.base += 1
return None
if not dx:
return 90 if dy > 0 else 270
if not dy:
return 0 if dx > 0 else 180
if dx * dy > 0:
return math.degrees(math.atan(dy/dx)) + (0 if dx > 0 else 180)
return math.degrees(math.atan(-dx/dy)) + (90 if dx < 0 else 270)
angles = []
for p in points:
degree = helper(p)
if degree is not None:
angles.append(degree)
angles.sort()
# 为了将第四象限和第一象限连贯起来作比较
angles = angles + [360 + a for a in angles]
l = r = ans = 0
while l < len(angles):
#窗口拓展
while r < len(angles) and angles[r] - angles[l] <= angle:
r += 1
ans = max(ans, r - l)
#窗口滑动
l += 1
return ans + self.base