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 <= 501 <= 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;}
}