如何在大小不同的两个向量中找到公共数

时间:2019-09-24 00:07:52

标签: c++ stl

我有两个大小不同的整数向量:

vector 1 = {1、1、3、3、3、5、5、8、8、8}

vector 2 = {1、3、3、5、8}

我试图遍历这两个向量,并比较其中的值以查看它们是否相似,然后将它们添加到新的向量中。

这是我尝试过的:

vector<int> firstList{1, 1, 3, 3, 3, 5, 5, 8, 8, 8}
vector<int> secondList{1, 3, 3, 5, 8}
vector<int> finalList;

    for (std::vector<char>::iterator i = firstList.begin(); i != firstList.end(); ++i)
    {
        if (std::find(secondList.begin(), secondList.end(), *i) != secondList.end())
        {
            finalList.push_back(*i);
        }
    }

我在finalList中期望的输出是:{1、3、3、5、8}

实际输出为:{1、1、3、3、3、5、5、8、8、8},当我只需要5时,它将返回10个值。

感谢您的帮助!

3 个答案:

答案 0 :(得分:1)

制作最小列表的副本,然后删除该列表上的找到的项目,以便在下一个循环中将排除找到的项目。

vector<int> firstList{1, 1, 3, 3, 3, 5, 5, 8, 8, 8}
vector<int> secondList{1, 3, 3, 5, 8}
vector<int> finalList;

vector<int> secondListCopy = secondList;

for (std::vector<int>::iterator i = firstList.begin(); i != firstList.end(); ++i)
{
    std::vector<int>::iterator i2 = std::find(secondListCopy.begin(), secondListCopy.end(), *i) ;
    if (i2 != secondListCopy.end())
    {
        finalList.push_back(*i);
        secondListCopy.erase(i2);

        if(secondListCopy.empty()) break; //just additional optimization.
    }
}

答案 1 :(得分:0)

您可以使用set。用最大的向量初始化集合。

using namespace std;

int main()
{
vector<int> firstList{1, 1, 3, 3, 3, 5, 5, 8, 8, 8};
vector<int> secondList{1, 3, 3, 5, 8};
vector<int> finalList;

set<int> firstSet{firstList.begin(),firstList.end()};
for(auto& i : secondList)
    if(*firstSet.find(i)) //find i in firstSet/firstList
       finalList.push_back(i);


for(auto& i : finalList)
   cout << i << " ";  //Output: 1 3 3 5 8 

}

答案 2 :(得分:0)

在O(NlogN)中执行此操作的最佳方法是对两个向量进行排序并同时循环它们。

sort(firstList.begin(),firstList.end());
sort(secondList.begin(),secondList.end());
int i = 0, j = 0;
while (i < firstList.size() && j < secondList.size())
{
    if (firstList[i] == secondList[j])
    {
        finalList.push_back(firstList[i]);
        i++;j++;
    }
    else if (firstList[i] < secondList[j])
    {
        i++;
    }
    else
    {
        j++;
    }
}