题目:
Given n points on a 2D plane, find if there is such a line parallel to y-axis that reflect the given points.
Example 1:
Given points = [[1,1],[-1,1]]
, return true
.
Example 2:
Given points = [[1,1],[-1,-1]]
, return false
.
Follow up:
Could you do better than O(n2)?
思路:
如果存在一条和y轴平行的对称线,那么该线的x坐标一定是点集中x的最大值和最小值的平均值。因此,我们首先遍历一遍点集,找出x的最大值和最小值,从而确定对称线;接着就对点集中的每个点,查找有没有关于该对称线对称的点。
这道题目其实也有需要clarify的地方,例如如果点集中有重合点怎么办?我们在这里假设点集中的重合点都算作一个点。因此我们采用set这种数据结构保存点集,便于快速查找。整个算法的时间复杂度是O(nlogn),空间复杂度是O(n)。
代码:
class Solution {
public:
bool isReflected(vector<pair<int, int>>& points) {
set<pair<int, int>> st;
int Min = INT_MAX, Max = INT_MIN;
for(auto val: points) {
Min = min(Min, val.first);
Max = max(Max, val.first);
st.insert(val);
}
int sum = Min + Max;
for(auto val: points) {
if(st.count(make_pair(sum - val.first, val.second)) == 0)
return false;
}
return true;
}
};