2010-12-07 34 views
2

当我从像载体的非嵌套容器删除,我做类似:擦除删除成语在嵌套容器中删除? (删除外部的; C++ STL)

struct is_to_remove 
{ 
    is_to_remove(dynamic_bitset<>& x) : x(x) {} 
    const bool operator()(unsigned int id) 
    { 
     return x[id]; 
    } 

private: 
    dynamic_bitset<> x; 
}; 

inline static void remove_elements_in_vector(vector<unsigned int>& vec, boost::dynamic_bitset<>& to_remove) 
{ 
    // use the erase-remove idiom to remove all elements marked in bitset 
    vec.erase(remove_if(vec.begin(), vec.end(), is_to_remove(to_remove)), vec.end()); 
} 

即所谓erase-remove idiom

现在,我有第二个数据结构vector<vector<unsigned int> >deque<vector<unsigned int> >,其中我想删除外部容器元素(它本身是一个内部类型的容器)根据bitset。

  • 是否可以在这个嵌套容器类型上使用擦除 - 删除习惯用法?
  • 如果是这样,它怎么可能?
  • 是否有限制? (比如:vec的vec是可能的,但不是vec的veque)?

我的第一个和天真的方法是以下。我假设remove_if是按顺序迭代的,并且依次遍历元素并依次决定。 这是一个错误的假设?

struct is_to_remove_new 
{ 
    is_to_remove_new(dynamic_bitset<>& x, unsigned int index) : x(x), index(index) {} 
    const bool operator()(vector<unsigned int> & vec) 
    { 
     return x[index++]; 
    } 

private: 
    dynamic_bitset<> x; 
    unsigned int index; 
}; 

inline static void remove_elements_in_vectorvector(vector<vector<unsigned int> >& vec, boost::dynamic_bitset<>& to_remove) 
{ 
    // use the erase-remove idiom to remove all elements marked in bitset 
    vec.erase(remove_if(vec.begin(), vec.end(), is_to_remove_new(to_remove, 0)), vec.end()); 
} 

结果是错误的,因此我在这里寻找一个正确的解决方案。我想我假设了一些事情,这是不能保证的。对我而言,基本问题如下:如何获取内部容器的身份以检查是否要删除它。
我上面发布的天真方法只是计数,并假设一个顺序处理。

感谢您的帮助。

的Sascha

更新和警告

对于矢量o载体,斯塔斯溶液工作很大。但我认为这个解决方案不适用于向量的转换,因为deque不是以连续的方式保存的。这意味着,仿函数中的索引计算失败。

任何人都可以验证吗?

回答

3

矢量中的哪些元素无关紧要。如果您定义哪些应该被删除,它们将被删除。

正如你所说,问题是如何识别向量中的元素。最明显的答案是其索引[0; vector_size - 1]

感谢一个向量,您可以很容易地通过元素本身获取元素索引。

std::vector<std::string> vec; 
vec.push_back("zero"); 
vec.push_back("one"); 
vec.push_back("two"); 

std::string& elem = vec[2]; 
int index = std::distance(&vec[0], &elem); // get element index by element itself 
// index == 2 

所以,你可以很容易地通过内部remove_if算法谓词的索引标识矢量元素。看看下面的例子。这是很愚蠢的,使用硬编码std::bitset<6>,但是这仅仅是一个例证:

#include <vector> 
#include <string> 
#include <bitset> 

struct ToRemove 
{ 
    ToRemove(std::vector<std::string>& vec, const std::string& mask) 
    : m_vec(vec) 
    , m_mask(mask) 
    {} 

    bool operator()(std::string& obj) 
    { 
     const int index = std::distance(&m_vec[0], &obj); 
     return m_mask[index]; 
    } 

    std::vector<std::string>& m_vec; 
    std::bitset<6> m_mask; 
}; 

使用

int main (int argc, char const *argv[]) 
{ 
    std::vector<std::string> vec; 
    vec.push_back("zero"); 
    vec.push_back("one"); 
    vec.push_back("two"); 
    vec.push_back("three"); 
    vec.push_back("four"); 
    vec.push_back("five"); 

    std::string mask = ("010011"); // Remove "zero", "one" and "four" 
    vec.erase(remove_if(vec.begin(), vec.end(), ToRemove(vec, mask)), vec.end()); 

    for (unsigned i = 0; i < vec.size(); ++i) 
    { 
     std::cout << vec[i] << " "; 
    } 
    std::cout << std::endl; 

    return 0; 
} 

结果

two three five 
+0

就像一个魅力。谢谢。接下来要做的事情是:用矢量 :-)做这件事。 – sascha 2010-12-07 22:46:59