[HashMap/Set][Easy] 2215. Find the Difference of Two Arrays

https://leetcode.com/problems/find-the-difference-of-two-arrays/description

Find the Difference of Two Arrays - LeetCode

 

Q. 주어진 2개의 입력 배열 num1, num2에 대해 2개의 조건을 만족하는 vector를 출력하라. 

 이때 첫번째 vector에는 num1에는 있지만 num2에는 없는 원소들 모음을, 

  2번째 vector에는 num2에는 있지만 num1에는 없는 원소들 모음을 중복 없이 출력하라. 

 

Solution. 

 hash를 쓰면 O(N)에 무난히 풀수 있는 문제. 

 

vector<vector<int>> findDifference(vector<int>& nums1, vector<int>& nums2)
{
    vector<vector<int>> vRet;
    unordered_set<int> present, sRet;
    for (int k = 0; k < nums2.size(); ++k)
    {
        if (present.find(nums2[k]) == present.end())
            present.insert(nums2[k]);
    }
    for (int k = 0; k < nums1.size(); ++k)
    {
        if (present.find(nums1[k]) == present.end())
        {
            if (sRet.find(nums1[k]) == sRet.end())
                sRet.insert(nums1[k]);
        }
    }
    vRet.push_back(vector<int>(sRet.begin(), sRet.end()));

    present.clear();
    sRet.clear();
    for (int k = 0; k < nums1.size(); ++k)
    {
        if (present.find(nums1[k]) == present.end())
            present.insert(nums1[k]);
    }
    for (int k = 0; k < nums2.size(); ++k)
    {
        if (present.find(nums2[k]) == present.end())
        {
            if (sRet.find(nums2[k]) == sRet.end())
                sRet.insert(nums2[k]);
        }
    }
    vRet.push_back(vector<int>(sRet.begin(), sRet.end()));
    return vRet;
}