题目:
给你一个下标从 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
提交通过: