5534. 可见点的最大数目(C++)---极角排序 + 滑动窗口 解题

题目详情

给你一个点数组 points 和一个表示角度的整数 angle ，你的位置是 location ，其中 location = [posx, posy] 且 points[i] = [xi, yi] 都表示 X-Y 平面上的整数坐标。

最开始，你面向东方进行观测。你 不能 进行移动改变位置，但可以通过 自转 调整观测角度。换句话说，posx 和 posy 不能改变。你的视野范围的角度用 angle 表示， 这决定了你观测任意方向时可以多宽。设 d 为逆时针旋转的度数，那么你的视野就是角度范围 [d - angle/2, d + angle/2] 所指示的那片区域。

对于每个点，如果由该点、你的位置以及从你的位置直接向东的方向形成的角度 位于你的视野中 ，那么你就可以看到它。

同一个坐标上可以有多个点。你所在的位置也可能存在一些点，但不管你的怎么旋转，总是可以看到这些点。同时，点不会阻碍你看到其他点。

返回你能看到的点的最大数目。

 

示例 1：

输入：points = [[2,1],[2,2],[3,3]], angle = 90, location = [1,1]
输出：3
解释：阴影区域代表你的视野。在你的视野中，所有的点都清晰可见，尽管 [2,2] 和 [3,3]在同一条直线上，你仍然可以看到 [3,3] 。

示例 2：

输入：points = [[2,1],[2,2],[3,4],[1,1]], angle = 90, location = [1,1]
输出：4
解释：在你的视野中，所有的点都清晰可见，包括你所在位置的那个点。

示例 3：

输入：points = [[0,1],[2,1]], angle = 13, location = [1,1]
输出：1
解释：如图所示，你只能看到两点之一。

 

提示：

• 1 <= points.length <= 105
• points[i].length == 2
• location.length == 2
• 0 <= angle < 360
• 0 <= posx, posy, xi, yi <= 109
  
   

题目链接

——题目难度:中等


 

---

将每个极角加 360 后 加到后面的原因是为了方便双指针处理特殊情况，下面引用一位力扣大佬的题解

因为按照第三象限 -> 第四象限 -> 第一象限 -> 第二象限的顺序，逆时排列，
所以v[r + 1] - v[l]就是相当于从 l 那个角逆时针到 r + 1 那个角的角度，且 v[r + 1] 在任何时刻都 >= v[l]

代码如下

class Solution {
public:
	int visiblePoints(vector<vector<int>>& points, int angle, vector<int>& location) {
		const double PI = 3.14159;
		int x = location[0], y = location[1], same = 0; //same用于统计相同点个数 
		vector<double> v; //用于保存极角
		for (int i = 0; i < points.size(); i++) {
			int tx = points[i][0], ty = points[i][1];
			if (x == tx && y == ty) {
				same++;
			}
			else {
				v.push_back(atan2(ty - y, tx - x) * 180 / PI);
			}
		}
		sort(v.begin(), v.end());
		int n = v.size(); //真正极角集的个数
        /*因为v存储的是角度，所以v增加2pi（转一圈）之后，并不影响相对位置，所以可以把v增加360后加入数组尾部*/
		for(int i = 0; i < n; i++) { //将每个极角加 360 后 加到后面
			v.push_back(v[i] + 360);
		}
		
		/*滑动窗口(双指针)*/
		/*将每个极角加 360 后 加到后面后的滑动窗口*/
		/*相当于L遍历了一遍极角集*/
		/*相当于r遍历了两遍极角集，只是两遍极角的比较方式有所改变*/
		int r = 0, ans = 1;
		for (int l = 0; l < n; l++) {
			while (r + 1 < v.size() && v[r + 1] - v[l] <= (double)angle) r++;
			ans = max(ans, r - l + 1);
		}

		return ans + same;
	}
};