有没有一种很好的方法来根据std :: set中的元素切片std :: vector？换句话说,std :: set中的元素在向量中保存了我想要的索引.当然,我可以用代码完成这个：
#include <set>
#include <vector>
#include <iostream>
#include <iterator>
template <typename T>
std::vector <T> slice(
std::vector <T> const & x,std::set <unsigned int> const & I
) {
auto y = std::vector <T> ();
for(auto const & i : I)
y.push_back(x[i]);
return y;
}
int main() {
auto x = std::vector <double> { 1.2,2.3,3.4,4.5,5.6};
auto I = std::set <unsigned int> { 0,3,4};
auto y = slice(x,I);
std::copy(y.begin(),y.end(),std::ostream_iterator <double>(
std::cout,"\n"));
}
哪个正确返回
1.2 4.5 5.6
然而,这感觉有点笨拙.有没有更好的办法？
解决方法
你可以通过简单地使用std :: transform来实现这个目的：
int main()
{
std::vector<double> x { 1.2,5.6};
std::set<unsigned int> I { 0,4};
std::vector<double> y(I.size());
std::transform(I.begin(),I.end(),y.begin(),[&x](unsigned int i) { return x[i]; });
std::copy(y.begin(),std::ostream_iterator <double>(std::cout,"\n"));
}
