程式語言 - LeetCode - C++ - 2215. Find the Difference of Two Arrays



題目:


解答:

class Solution {
public:
    vector<vector<int>> findDifference(vector<int>& nums1, vector<int>& nums2) {
        std::set<int> n1;
        std::set<int> n2;
        std::vector<std::vector<int>> r;

        for (std::vector<int>::iterator it=nums1.begin(); it != nums1.end(); it++) {
            if (std::find(nums2.begin(), nums2.end(), *it) == nums2.end()) {
                n1.insert(*it);
            }
        }

        for (std::vector<int>::iterator it=nums2.begin(); it != nums2.end(); it++) {
            if (std::find(nums1.begin(), nums1.end(), *it) == nums1.end()) {
                n2.insert(*it);
            }
        }

        r.push_back(std::vector<int>(n1.begin(), n1.end()));
        r.push_back(std::vector<int>(n2.begin(), n2.end()));
        return r;
    }
};