力扣算法题-2013. 检测正方形 Java实现

题目

给你一个在 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

首先确定点的存储数据结构和数据存储方式，为了便于检索点是否存在，确定数据结构类型为嵌套hashmap数据存储类型。
其中设点为（x,y）集合中的存储合适为 (y,(x,该点的个数))

添加流程：
节点添加到存储，需要将节点添加到如上设置的数据结构的集合中去；

计算流程：
设需要计算的点为（x,y）
设对应的正方形的另一个纵轴值为 col，则正方形的边长为d = col - y
则其他的三个点为(x,col),(x+d,col),(x+d,y)
或者为(x,col),(x-d,col),(x-d,y)

则点(x,y)对应的正方形的个数为
点(x,col)的个数*(x+d,col)个数*(x+d,y) +点(x,col)的个数*(x-d,col)个数*(x-d,y)

程序

class DetectSquares {
    Map<Integer, Map<Integer, Integer>> map;

    public DetectSquares() {
        map = new HashMap<Integer, Map<Integer, Integer>>();
    }
    /**集合为(key,(key,value))格式，存储(y,(x,该点的个数)) */
    public void add(int[] point) {
        int x = point[0], y = point[1];
        /**若已经有 y 的键，则不再覆盖已有键值对 */
        map.putIfAbsent(y, new HashMap<Integer, Integer>());
        /**获取 y 键的值对应的集合 */
        Map<Integer, Integer> yCnt = map.get(y);
        /**将点的x值放入集合，并计算个数 */
        yCnt.put(x, yCnt.getOrDefault(x,0)+1);
    }
    
    public int count(int[] point) {
        int cnt = 0 ;
        int x = point[0], y = point[1];
        /**若不存在同纵轴的点，则返回0 */
        if(!map.containsKey(y)){
            return 0;
        }
        /**获取纵轴为y的集合 */
        Map<Integer, Integer> yCnt = map.get(y);
        /**获取所有节点的集合 */
        Set<Map.Entry<Integer, Map<Integer, Integer>>> entries = map.entrySet();
        /**遍历所有节点的集合 */
        for(Map.Entry<Integer, Map<Integer, Integer>> entry:entries){
            /**获取纵轴值 */
            int col = entry.getKey();
            /**获取纵轴值col 对应的横轴集合 */
            Map<Integer, Integer> colCnt = entry.getValue();
            /**纵轴相同的不算，跳过四点相同的场景 */
            if(col!=y){
                /**获取正方形的边长 */
                int d = col - y;
                /**获取点左右侧的正方形个数 */
                cnt += colCnt.getOrDefault(x,0)*yCnt.getOrDefault(x+d,0)*colCnt.getOrDefault(x+d,0);
                cnt += colCnt.getOrDefault(x,0)*yCnt.getOrDefault(x-d,0)*colCnt.getOrDefault(x-d,0);
            }
        }
        return cnt;
    }
}

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