寻找在另一个映射中的向量的公共元素

finding the common elements of vectors which are inside another map

本文关键字：向量 元素 另一个 映射 寻找      更新时间：2023-10-16

我有一个地图，定义为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;

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

如果你能保证你的向量是排序的，那么你可以大大提高你的性能，从0 (mn)到O(m + n):

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

你完全把测试弄反了:

if(any_in_2nd_vector_iterator==2nd_vector_end){
当元素在第二个向量中没有找到时

为真。试试!=

然后，可以使用引用来避免不必要的vector复制:

const vector<int>& first_vector=my_map[s1];

，如果你的向量变得更大，它可能会更快地做一个哈希查找(或预排序合并，如果向量总是保持排序)，而不是调用find必须检查每个元素。

相关文章：