leetcode-每日一题2022.1.26 检测正方形

最新推荐文章于 2022-12-01 17:12:09 发布

还记得樱花正开~

最新推荐文章于 2022-12-01 17:12:09 发布

阅读量871

点赞数

分类专栏： leetcode 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/ZHUYAN1209/article/details/122713414

版权

leetcode 专栏收录该内容

223 篇文章 2 订阅

订阅专栏

题目

力扣

思路哈希表

用哈希表再套一个哈希表，外层哈希表的键是行的纵坐标，内层哈希表的键是列的横坐标，值是该位置点的个数。count函数的实现，先判断y是不是在哈希表里，如果不在直接返回0。再遍历哈希表的所有行，计算出正方形的边长，让结果加上另外三个点出现次数的乘积，有向左取和向右取两种情况。

代码

class DetectSquares:

    def __init__(self):
        self.cnt = defaultdict(Counter)


    def add(self, point: List[int]) -> None:
        x, y = point
        self.cnt[y][x] += 1

    def count(self, point: List[int]) -> int:
        x, y = point
        if(not y in self.cnt):
            return 0
        yCnt = self.cnt[y]
        ans = 0
        for col,colCnt in self.cnt.items():
            if(col == y):
                continue
            d =  col - y
            ans += colCnt[x]*yCnt[x-d]*colCnt[x-d]
            ans += colCnt[x]*yCnt[x+d]*colCnt[x+d]
        return ans


# Your DetectSquares object will be instantiated and called as such:
# obj = DetectSquares()
# obj.add(point)
# param_2 = obj.count(point)