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

原创

freesan44

修改于 2021-09-03 10:59:04

3610

修改于 2021-09-03 10:59:04

文章被收录于专栏：freesan44

题目

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

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

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

请你返回一个数组 answer ，其中 answerj是第 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)

原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。

如有侵权，请联系 cloudcommunity@tencent.com 删除。

数据结构