Featured image of post LeetCode 75系列 - 2215. 找出两数组的不同

LeetCode 75系列 - 2215. 找出两数组的不同

题目描述

给你两个下标从 0 开始的整数数组 nums1nums2 ,请你返回一个长度为 2 的列表 answer ,其中:

  • answer[0]nums1 中所有 存在于 nums2 中的 不同 整数组成的列表。
  • answer[1]nums2 中所有 存在于 nums1 中的 不同 整数组成的列表。

**注意:**列表中的整数可以按 任意 顺序返回。

示例 1:

1
2
3
4
5
输入:nums1 = [1,2,3], nums2 = [2,4,6]
输出:[[1,3],[4,6]]
解释:
对于 nums1 ,nums1[1] = 2 出现在 nums2 中下标 0 处,然而 nums1[0] = 1 和 nums1[2] = 3 没有出现在 nums2 中。因此,answer[0] = [1,3]。
对于 nums2 ,nums2[0] = 2 出现在 nums1 中下标 1 处,然而 nums2[1] = 4 和 nums2[2] = 6 没有出现在 nums2 中。因此,answer[1] = [4,6]。

示例 2:

1
2
3
4
5
输入:nums1 = [1,2,3,3], nums2 = [1,1,2,2]
输出:[[3],[]]
解释:
对于 nums1 ,nums1[2] 和 nums1[3] 没有出现在 nums2 中。由于 nums1[2] == nums1[3] ,二者的值只需要在 answer[0] 中出现一次,故 answer[0] = [3]。
nums2 中的每个整数都在 nums1 中出现,因此,answer[1] = [] 。 

提示:

  • 1 <= nums1.length, nums2.length <= 1000
  • -1000 <= nums1[i], nums2[i] <= 1000

题解

通过 set 容器进行去重,然后利用 set 容器的find()或者count()函数判断是否存在该元素,不存在则放入vector 中。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution {
public:
    vector<vector<int>> findDifference(vector<int>& nums1, vector<int>& nums2) {
        set<int> s1;
        set<int> s2;
        vector<int> r1;
        vector<int> r2;
        for (int i = 0; i < nums1.size(); ++i) {
            s1.insert(nums1[i]);
        }
        for (int i = 0; i < nums2.size(); ++i) {
            s2.insert(nums2[i]);
        }
        for (int num : s1) {
            if (s2.count(num) == 0) {
                r1.push_back(num);
            }
        }
        for (int num : s2) {
            if (s1.count(num) == 0) {
                r2.push_back(num);
            }
        }

        return {r1, r2};
    }
};