C ++向量被另一个向量索引

时间:2019-03-05 14:09:25

标签: c++ vector indexing

很想问这个显然很简单的问题,但是我无法找到我的问题的确切答案,而且我显然没有经验,无法从类似问题中得出答案...

假设我有以下情况

#include <vector>

int main() {
    std::cout << "Test" << std::endl;

    int myints1[] = {1, 1 , 0, 0};
    std::vector<int> vec1 (myints1, myints1 + sizeof(myints) / sizeof(int) );
    int myints2[] = {1, 2 , 3, 4};
    std::vector<int> vec2 (myints, myints2 + sizeof(myints2) / sizeof(int) );

    std::vector<int> resultVec;

    // Now as a result I want to get the resultVec as all entries in vec2 
    // where vec1 == 0, resulting in resultVec = [3,4]
    return 0

}

如何通过另一个向量的值选择一个向量的所有条目?

2 个答案:

答案 0 :(得分:6)

只需在索引上循环就可以了。该代码假定vec2的大小至少与vec1一样大。

int main() {
    std::cout << "Test" << std::endl;

    int myints1[] = {1, 1 , 0, 0};
    std::vector<int> vec1 (myints1, myints1 + sizeof(myints) / sizeof(int) );
    int myints2[] = {1, 2 , 3, 4};
    std::vector<int> vec2 (myints, myints2 + sizeof(myints2) / sizeof(int) );

    std::vector<int> resultVec;

    for (unsigned i = 0; i < vec1.size(); ++i) {
        if (!vec1[i]) resultVec.push_back(vec2[i]);
    }
    return 0

}

答案 1 :(得分:6)

您可以有一个使两个迭代器递增的循环。因为您的集合是相同的类型,所以您可以在for中将它们除法,并且由于它们的大小是相同的,因此您只需要检查一个即可。

for (auto it1 = vec1.begin(), it2 = vec2.begin(); it1 != vec1.end(); ++it1, ++it2) 
{
     // Use *it1 and *it2
}

如果您可以使用C ++ 17和boost,则可以使用范围很广的

for (auto & [val1, val2] : boost::combine(vec1, vec2))
{
     // Use val1 and val2
}

请注意,您可以使用{}来初始化vector,所以我不会为myints所困扰。

std::vector<int> vec1 = {1, 1, 0, 0};
std::vector<int> vec2 = {1, 2, 3, 4};