c++ - 找到另一个 map 内 vector 的公共(public)元素

Question

亲爱的大家， 我有一张 map ，定义为 map<int, vector<int> > my_map .因此，例如它看起来如下所示，

my_map=
0 1 2 6 5 4
1 0 2 3 4 5
2 0 1 4
3 1
4 1 2 0 7 5 6
5 1 0 6 4
6 0 5 4
7 4

vector<int> element_common(map<int, vector<int> > &my_map, int s1, int s2){
     vector<int> common2;
     vector<int> first_vector=my_map[s1];     
     vector<int>::iterator snd_vector_begin= my_map[s2].begin();
     vector<int>::iterator snd_vector_end= my_map[s2].end();          
     for (vector<int>::iterator any=first_vector.begin();
                            any!=first_vector.end();
                            any++){            
         // check if any is in other_list
         vector<int>::iterator any_in_snd_vector_iterator= find (snd_vector_begin,snd_vector_end, *any);            
         if(any_in_snd_vector_iterator==snd_vector_end){
            common2.push_back(*any); 
            }
         }
     return common2;

我在函数上方编写了一个函数，以从与 2 个给定键相关的 vector 部分获取公共(public)元素。当键等于 s1 和 s2 时，即 s1=1 和 s2=4。那么我的函数应该给我 0,2,5。但我得到了 3,4。请帮助我纠正我的功能。

Answer 1

如果您可以保证您的 vector 已排序，那么您可以将性能从 O(mn) 显着提高到 O(m + n):

std::vector<int> common;
std::set_intersection(vec1.begin(), vec1.end(),
                      vec2.begin(), vec2.end(),
                      std::back_inserter(common));