Krononberg

349. Intersection of Two Arrays.cpp 본문

개발 로그/알고리즘

349. Intersection of Two Arrays.cpp

k._. 2021. 11. 20. 14:57

✔ set 사용 연습. 범위기반 for loop 편하니 익숙해지자.

 

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int>res;
        set<int>s;
        for(int i=0; i<nums1.size(); i++){
            s.insert(nums1[i]);
        }
        // for(auto it = s.begin(); it!=s.end(); it++){
        //     for(int i =0; i<nums2.size(); i++){
        //         if(*it == nums2[i]){
        //             res.push_back(*it);
        //             break;   
        //         }
        //     }
        // }
        for(auto ele : s){
            for(auto num : nums2){
                if(ele ==num){
                    res.push_back(ele);
                    break;
                }
            }
        }
        return res;
    }
};