找出缺失和重复的数字

问题

给你一个下标从 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] 。

思想

使用map记录每个值在数组中出现的次数，然后遍历map找出符合条件的值。

代码

class Solution {
    public int[] findMissingAndRepeatedValues(int[][] grid) {
        int n = grid.length;
        int right = n * n;
        int[] ans = new int[2];
        HashMap<Integer,Integer> map = new HashMap<>();
        for(int i =1; i <= right; i++){
            map.put(i,0);
        }
        for(int i = 0; i < n; i++){
            for(int j = 0; j < n; j++){  
                 map.put(grid[i][j],map.get(grid[i][j]) + 1);
            }
        }
        for(Integer key : map.keySet()){
            if(map.get(key) == 2 && key != 0){
                ans[0] = key;
            }else if(map.get(key) == 0){
                ans[1] = key;
                if(ans[0] != 0){
                     break;
                } 
            }
        }
        return ans;
    }
}