LeetCode
找出缺失的重复数字
题目链接:2965. 找出缺失和重复的数字 - 力扣(LeetCode)
题目描述
给你一个下标从 0 开始的二维整数矩阵 grid
,大小为 n * n
,其中的值在 [1, n2]
范围内。除了 a
出现 两次,b
缺失 之外,每个整数都 恰好出现一次 。
任务是找出重复的数字a
和缺失的数字 b
。
返回一个下标从 0 开始、长度为 2
的整数数组 ans
,其中 ans[0]
等于 a
,ans[1]
等于 b
。
示例 1:
输入:grid = [[1,3],[2,2]]
输出:[2,4]
解释:数字 2 重复,数字 4 缺失,所以答案是 [2,4] 。
示例 2:
输入:grid = [[9,1,7],[8,9,2],[3,4,6]]
输出:[9,5]
解释:数字 9 重复,数字 5 缺失,所以答案是 [9,5] 。
提示:
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
。
思路
用数组统计每个数出现的次数,寻找出现两次的数,和出现零次的数
代码
C++
class Solution {
public:
vector<int> findMissingAndRepeatedValues(vector<vector<int>>& grid) {
int n = grid.size();
vector<int> cnt(n * n + 1);
for (auto& row : grid) {
for (int x : row) {
cnt[x]++;
}
}
vector<int> ans(2);
for (int i = 1; i <= n * n; i++) {
if (cnt[i] == 2) {
ans[0] = i; // 出现两次的数
} else if (cnt[i] == 0) {
ans[1] = i; // 出现零次的数
}
}
return ans;
}
};
Java
public class Solution {
public int[] findMissingAndRepeatedValues(int[][] grid) {
int n = grid.length;
int[] cnt = new int[n * n + 1];
for (int[] row : grid) {
for (int x : row) {
cnt[x]++;
}
}
int[] ans = new int[2];
for (int i = 1; i <= n * n; i++) {
if (cnt[i] == 2) {
ans[0] = i;
} else if (cnt[i] == 0) {
ans[1] = i;
}
}
return ans;
}
}