LeetCode 1828. 统计一个圆中点的数目

作者: freesan44 | 来源:发表于2021-09-22 07:56 被阅读0次

题目

给你一个数组 points ,其中 points[i] = [xi, yi] ,表示第 i 个点在二维平面上的坐标。多个点可能会有 相同 的坐标。

同时给你一个数组 queries ,其中 queries[j] = [xj, yj, rj] ,表示一个圆心在 (xj, yj) 且半径为 rj 的圆。

对于每一个查询 queries[j] ,计算在第 j 个圆 内 点的数目。如果一个点在圆的 边界上 ,我们同样认为它在圆 内 。

请你返回一个数组 answer ,其中 answer[j]是第 j 个查询的答案。

示例 1:



输入:points = [[1,3],[3,3],[5,3],[2,2]], queries = [[2,3,1],[4,3,1],[1,1,2]]
输出:[3,2,2]
解释:所有的点和圆如上图所示。
queries[0] 是绿色的圆,queries[1] 是红色的圆,queries[2] 是蓝色的圆。

示例 2:



输入:points = [[1,1],[2,2],[3,3],[4,4],[5,5]], queries = [[1,2,2],[2,2,2],[4,3,2],[4,3,3]]
输出:[2,3,2,4]
解释:所有的点和圆如上图所示。
queries[0] 是绿色的圆,queries[1] 是红色的圆,queries[2] 是蓝色的圆,queries[3] 是紫色的圆。

解题思路

class Solution:
    def countPoints(self, points: List[List[int]], queries: List[List[int]]) -> List[int]:
        resArr = []
        # 针对每个圆
        for quer in queries:
            res = 0
            for point in points:
                a, b = point[0], point[1]
                X, Y, R = quer[0], quer[1], quer[2]
                # (X-a)^2+(Y-b)^2=r^2
                if (X-a)**2 + (Y-b)**2 <= R**2:
                    res += 1
            resArr.append(res)
        return resArr

if __name__ == '__main__':
    points = [[1,3],[3,3],[5,3],[2,2]]
    queries = [[2,3,1],[4,3,1],[1,1,2]]
    ret = Solution().countPoints(points, queries)
    print(ret)

相关文章

网友评论

    本文标题:LeetCode 1828. 统计一个圆中点的数目

    本文链接:https://www.haomeiwen.com/subject/xhihwltx.html