有 2 个未排序的 int v1 和 v2 向量,其中 v1 包含 v2 的子集
v1: 8 12 4 17
v2: 6 4 14 17 9 0 5 12 8
有什么办法,如何用 v2 中的位置索引替换 v1 的项目?
v1: 8 7 1 3
用 2 为循环编写这样的算法是没有问题的......
但是有没有使用 std::transform 的解决方案?
结合std::transform
调用的函数对象std::find
:
#include <vector>
#include <algorithm>
#include <iostream>
#include <iterator>
struct find_functor
{
std::vector<int> &haystack;
find_functor(std::vector<int> &haystack)
: haystack(haystack)
{}
int operator()(int needle)
{
return std::find(haystack.begin(), haystack.end(), needle) - haystack.begin();
}
};
int main()
{
std::vector<int> v1 = {8, 12, 4, 17};
std::vector<int> v2 = {6, 4, 14, 17, 9, 0, 5, 12, 8};
// in c++11:
std::transform(v1.begin(), v1.end(), v1.begin(), [&v2](int x){
return std::find(v2.begin(), v2.end(), x) - v2.begin();
});
// in c++03:
std::transform(v1.begin(), v1.end(), v1.begin(), find_functor(v2));
std::cout << "v1: ";
std::copy(v1.begin(), v1.end(), std::ostream_iterator<int>(std::cout, " "));
std::cout << std::endl;
return 0;
}
输出:
$ g++ -std=c++0x test.cpp
$ ./a.out
v1: 8 7 1 3
您可以std::find()
在转换中使用:
std::transform(v1.begin(), v1.end(), v1.begin(), [&](int v)->int {
return std::find(v2.begin(), v2.end(), v) - v2.begin());
});
std::transform
接受一个类似一元函数的对象。因此,您可以创建一个高效执行此操作的仿函数类,使用第二个向量构造它,然后将该仿函数应用于第一个向量。
template <typename T>
class IndexSeeker{
private:
map<T, int> indexes;
public:
IndexSeeker(vector<T> source){
for(int k = 0; k < t.size(); ++k){
indexes[source[k]] = k;
}
}
int operator()(const T& locateme){
if(indexes.find(T) != indexes.end()){
return indexes[T];
}
return -1;
}
}
通过将整个第二个列表缓存到地图中,查找索引是有效的,而不需要线性搜索。这要求类型 T 是可排序的(因此是可映射的)。如果 T 不可排序,则需要一种需要蛮力搜索的效率较低的方法。