2010-12-07 57 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