力扣2965. 找出缺失和重复的数字

题目：

        给你一个下标从 0 开始的二维整数矩阵 grid，大小为 n * n ，其中的值在 [1, n²] 范围内。除了 a 出现两次，b 缺失 之外，每个整数都恰好出现一次 。

        任务是找出重复的数字a 和缺失的数字 b 。

        返回一个下标从 0 开始、长度为 2 的整数数组 ans，其中 ans[0] 等于a，ans[1] 等于b。

提示：

• 2 <= n == grid.length == grid[i].length <= 50
• 1 <= grid[i][j] <= n * n
• 对于所有满足1 <= x <= n * n 的 x ，恰好存在一个 x 与矩阵中的任何成员都不相等。
• 对于所有满足1 <= x <= n * n 的 x ，恰好存在一个 x 与矩阵中的两个成员相等。
• 除上述的两个之外，对于所有满足1 <= x <= n * n 的 x ，都恰好存在一对 i, j 满足 0 <= i, j <= n - 1 且 grid[i][j] == x 。

思路：

用字典记录从1到n²的每个值出现的次数，键值对为[数字：出现次数]。然后在字典中找到值为0（缺失）或2（重复）对应的键即可。代码如下：

class Solution:
    def findMissingAndRepeatedValues(self, grid: List[List[int]]) -> List[int]:
        n = len(grid)
        ans = [0, 0]
        num_dict = {i:0 for i in range(1, n*n+1)}   # 数字：出现次数
        # 遍历矩阵，更新字典中的值
        for row in grid:
            for num in row:
                num_dict[num] += 1
        # 找到字典中值为0或2对应的键
        for key, values in num_dict.items():
            if values == 2:
                ans[0] = key
            if values == 0:
                ans[1] =key
        return ans

提交通过：