C++ 使用copy_if获得数组vector掩膜

假设给定一个数组vector<double> veca以及对应的掩膜（即指示标志数组）vector<bool> flags，获得veca中对应flags中为true的元素。

假设veca为{0.1, 0.2, 0.3, 0.4}，flags为{true, false, false, true}，则vecb应该为{0.1, 0.4}。

使用for循环自然可以很简单地解决这个问题，但是想要用标准模板库中的算法实现需要使用copy_if。使用copy算法的复制拷贝效率要比for的效率要高一些。

copy_if的由四个参数，前两个是输入元素的迭代器，拷贝两个迭代器之间的元素，第三个将元素拷贝到的位置，第四个是选择条件，即只拷贝改条件返回true的元素。

#include <vector>
#include <algorithm>
#include <iostream>
#include <numeric>    // accumulate
#include <iterator>

int main()
{
	std::vector<bool> flags{ true, true, false, true};
	std::vector<double> veca{ 0.1, 0.2, 0.3, -0.1 };
	
	std::vector<double> vecb;
	vecb.reserve(std::accumulate(flags.begin(), flags.end(), 0));
	
	// vecb在reserve之后并没有未元素分配内存，插入应该使用back_inserter(vecb)
	// vecb在resize之后为元素分配了内存，使用back_inserter(vecb)会在已经分配内存的元素之后插入
	// 这时应该使用 vecb.begin()，对已经分配的内存进行覆盖
	size_t i = 0;
	std::copy_if(veca.begin(), veca.end(), std::back_inserter(vecb),
                [&i, &flags](double a){return flags[i++]; });
	
	for (auto &s : vecb)
		std::cout << s << std::endl;

// 	i = 0;
// 	std::vector<double> vecc;
// 	std::remove_copy_if(veca.begin(), veca.end(), std::back_inserter(vecc),
// 		[&i, &flags](double a){return flags[i++]; });
// 	std::cout << "veca\n";
// 	for (auto& s : veca)
// 		std::cout << s << "\t";
// 	std::cout << "\nvecc\n";
// 	for (auto& s : vecc)
// 		std::cout << s << "\t";
	return 0;
}

使用lambda表达式作为第四个元素，捕获掩膜数组flags和元素序号i。
注意lambda表达式中flags和i都是使用的引用捕获，对于数组等数据结构要使用引用捕获，而且我们希望在lambda表达式中改变i的值，因此i也要是用引用捕获。
关于C++ lambda表达式更加详细的说明可参考博客和博客。

要注意的是reserve与resize的不同。

• reserve只调整数组的capacity，并不分配元素分配内存，因此需要则copy_if的第三个参数使用back_insterter；
• resize为元素分配好了空间，如果在copy_if的第三个参数使用back_inserter，则会在已经分配内存的元素后面插入，正确地做法是使用vecb.begin()，对已经分配内存的元素进行覆盖。

remove_copy_if感觉和copy_if差不多，就是逻辑相反，使用remove_copy_if会在第三个参数的位置上收集到第四个参数返回false的元素。具体差别还要在继续查查资料。