力扣刷题--2965. 找出缺失和重复的数字【简单】

码农小苏242024-05-29 18:08

题目描述

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

算法分析

先把二维数组的值全部放入一个一维数组当中，然后排序，如果相邻元素相等，就是重复的数字，然后遍历1-n*n中的数字，看他是否在一维数组当中。

完整代码

cpp 复制代码

class Solution {
public:
    vector<int> findMissingAndRepeatedValues(vector<vector<int>>& grid) {
        vector<int>arr;
        vector<int>ans;
        int n=grid.size();
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
                arr.push_back(grid[i][j]);  
        }
        sort(arr.begin(),arr.end());
        //重复的数字
        for(int i=0;i+1<arr.size();i++)
        {
            if(arr[i]==arr[i+1])
                ans.push_back(arr[i]);
        }
        //丢失的数字
        for(int i=1;i<=n*n;i++)
        {
            if(find(arr.begin(),arr.end(),i)==arr.end())
                ans.push_back(i);
        }
        return ans;
    }
};