找到最近的有相同 X 或 Y 坐标的点

106 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第10天,点击查看活动详情

1779. 找到最近的有相同 X 或 Y 坐标的点

难度简单73

给你两个整数 x 和 y ,表示你在一个笛卡尔坐标系下的 (x, y) 处。同时,在同一个坐标系下给你一个数组 points ,其中 points[i] = [ai, bi] 表示在 (ai, bi) 处有一个点。当一个点与你所在的位置有相同的 x 坐标或者相同的 y 坐标时,我们称这个点是 有效的 。

请返回距离你当前位置 曼哈顿距离 最近的 有效 点的下标(下标从 0 开始)。如果有多个最近的有效点,请返回下标 最小 的一个。如果没有有效点,请返回 -1 。

两个点 (x1, y1) 和 (x2, y2) 之间的 曼哈顿距离 为 abs(x1 - x2) + abs(y1 - y2) 。

 

示例 1:

输入: x = 3, y = 4, points = [[1,2],[3,1],[2,4],[2,3],[4,4]]
输出: 2
解释: 所有点中,[3,1][2,4][4,4] 是有效点。有效点中,[2,4][4,4] 距离你当前位置的曼哈顿距离最小,都为 1[2,4] 的下标最小,所以返回 2

示例 2:

输入: x = 3, y = 4, points = [[3,4]]
输出: 0
提示: 答案可以与你当前所在位置坐标相同。

示例 3:

输入: x = 3, y = 4, points = [[2,3]]
输出: -1
解释: 没有 有效点。

我们可以枚举数组 \textit{points}points 中所有的点并计算出答案。

当我们枚举到点 (px,py),如果 x=px,那么这两个点有相同的 xx 坐标,我们可以用距离 |y - \textit{py}|∣y−py∣ 更新答案;如果 y=\textit{py}y=py,那么这两个点有相同的 yy 坐标,我们可以用距离 |x - \textit{px}|∣x−px∣ 更新答案。

题目要求返回下标最小的一个最近有效点,我们只需要按照数据枚举点,在距离严格变小时才选择更新答案即可。

class Solution {
public:
    int nearestValidPoint(int x, int y, vector<vector<int>>& points) {
        int n = points.size();
        int best = numeric_limits<int>::max(), bestid = -1;
        for (int i = 0; i < n; ++i) {
            int px = points[i][0], py = points[i][1];
            if (x == px) {
                if (int dist = abs(y - py); dist < best) {
                    best = dist;
                    bestid = i;
                }
            }
            else if (y == py) {
                if (int dist = abs(x - px); dist < best) {
                    best = dist;
                    bestid = i;
                }
            }
        }
        return bestid;
    }
};