检测正方形

给你一个在 X-Y 平面上的点构成的数据流。设计一个满足下述要求的算法：

添加 一个在数据流中的新点到某个数据结构中。可以添加 重复 的点，并会视作不同的点进行处理。
给你一个查询点，请你从数据结构中选出三个点，使这三个点和查询点一同构成一个 面积为正 的 轴对齐正方形 ，统计 满足该要求的方案数目。
轴对齐正方形 是一个正方形，除四条边长度相同外，还满足每条边都与 x-轴 或 y-轴 平行或垂直。

实现 DetectSquares 类：

DetectSquares() 使用空数据结构初始化对象
void add(int[] point) 向数据结构添加一个新的点 point = [x, y]
int count(int[] point) 统计按上述方式与点 point = [x, y] 共同构造 轴对齐正方形 的方案数。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/detect-squares
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class DetectSquares {
    int count[][];
    public DetectSquares() {
        //记录坐标点为i,j的数目
        count=new int[1005][1005];
    }
    public void add(int[] point) {
        //添加各i,j数目
        count[point[0]][point[1]]++;
    }
    public int count(int[] point) {
        int ans=0;
        //计算斜率为-1的坐标和
        int a=point[0]+point[1];
        //计算斜率为1的坐标和
        int b=point[0]-point[1];
        for(int i=0;i<=1000;i++)
            {
                //比较i与xi的坐标
                if(i==point[0])
                    continue;
            int y1=a-i;
            int y2=i-b;
            //记录可与查询点形成矩形的其余三点的数目
            if(y1>=0&&y1<=1000)
                ans+=count[point[0]][y1]*count[i][point[1]]*count[i][y1];
            if(y2>=0&&y2<=1000)
                ans+=count[point[0]][y2]*count[i][point[1]]*count[i][y2];
            }
            return ans;
    }
}

/**
 * Your DetectSquares object will be instantiated and called as such:
 * DetectSquares obj = new DetectSquares();
 * obj.add(point);
 * int param_2 = obj.count(point);
 */