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

输入：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] 是蓝色的圆。

输入：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] 是紫色的圆。
 
提示：
1 <= points.length <= 500
points[i].length == 2
0 <= x​​​​​​i, y​​​​​​i <= 500
1 <= queries.length <= 500
queries[j].length == 3
0 <= xj, yj <= 500
1 <= rj <= 500
所有的坐标都是整数。

class Solution {
public:
    vector<int> countPoints(vector<vector<int>>& p, vector<vector<int>>& q) {
    	int n = q.size();
    	vector<int> ans(n, 0);
    	for(int i = 0; i < n; i++)
    	{
    		int x0 = q[i][0], y0 = q[i][1], r = q[i][2];
    		for(auto pi : p)
    		{
    			if((pi[0]-x0)*(pi[0]-x0)+(pi[1]-y0)*(pi[1]-y0) <= r*r)
    				ans[i]++; // 点在圆内
    		}
    	}
    	return ans;
    }
};