创建STL映射键迭代器

时间:2011-10-05 20:30:41

标签: c++ stl

通常,您有一个类似map<string,X>的地图,其中键是映射值的名称,您需要一个API,让消费者可以看到所有名称...以填充GUI列表框,例如。 您可以构建一个向量并将其作为API调用返回,但这样效率很低。您可以只返回对地图的引用,但随后也可以访问这些值,您可能不希望这样。

那么你怎么能编写一个兼容的类KeyIterator,它包装map并提供对该map中键的标准迭代器访问。

e.g:

map<string,X> m= ...
KeyIterator<string> ki(m);
for(KeyIterator<string>::iterator it=ki.begin();it!=ki.end();++it)
 cout << *it;

KeyIterator应该是轻量级的,因此您可以从几乎没有开销的方法返回它。

修改 我不确定我是否完美解释过,让我给出一个更好的用例(半伪):

class PersonManager
{
 private:
  map<string,Person> people;
 public:
  //this version has to iterate the map, build a new structure and return a copy
  vector<string> getNamesStandard();

  //this version returns a lightweight container which can be iterated
  //and directly wraps the map, allowing access to the keys
  KeyIterator<string> getNames();
};

void PrintNames(PersonManager &pm)
{
 KeyIterator<string> names = pm.getNames();
 for(KeyIterator<string>::iterator it=names.begin();it!=names.end();++it)
  cout << *it << endl;
}

3 个答案:

答案 0 :(得分:3)

template<typename iterator_type>
class KeyIterator
{
    iterator_type iterator;
public:
    typedef typename std::iterator_traits<iterator_type>::value_type::first_type value_type;
    KeyIterator(iterator_type i) : iterator(i) {}
    value_type operator*() { return iterator->first; }
    KeyIterator & operator++() { ++iterator; return *this; }
    bool operator!=(const KeyIterator & right) const { return iterator != right.iterator; }
    // ...
};

编辑:看到您的编辑后,我发现这并不是您要求的。你把你的类称为KeyIterator让我感到困惑,更恰当的名字是KeyContainer。您将无法仅在键类型上对其进行模板化,因为它必须包含对地图的某种引用;你需要完整的地图定义。

您的请求会使问题过于复杂,因为您必须定义两种不同的类型KeyIteratorKeyIterator::iterator

以下是使用我的班级的示例代码:

class PersonManager
{
private:
    map<string,Person> people;
public:
    //this version has to iterate the map, build a new structure and return a copy 
    vector<string> getNamesStandard(); 

    //this version returns a lightweight container which can be iterated 
    //and directly wraps the map, allowing access to the keys 
    KeyIterator<map<string,Person>::iterator> getNamesBegin(); 
    KeyIterator<map<string,Person>::iterator> getNamesEnd(); 
}; 

void PrintNames(PersonManager &pm) 
{ 
    KeyIterator<map<string,Person>::iterator> it = pm.getNamesBegin();
    KeyIterator<map<string,Person>::iterator> end = pm.getNamesEnd();
    for(it; it!=end; ++it) 
        cout << *it << endl; 
}

答案 1 :(得分:3)

#include <map>
#include <string>
#include <iterator>

template <class map>
class KeyIterator { 
    typename map::const_iterator iter_;
public:
    KeyIterator() {}
    KeyIterator(typename map::iterator iter) :iter_(iter) {}
    KeyIterator(typename map::const_iterator iter) :iter_(iter) {}
    KeyIterator(const KeyIterator& b) :iter_(b.iter_) {}
    KeyIterator& operator=(const KeyIterator& b) {iter_ = b.iter_; return *this;}
    KeyIterator& operator++() {++iter_; return *this;}
    KeyIterator operator++(int) {return KeyIterator(iter_++);}
    const typename map::key_type& operator*() {return iter_->first;}
    bool operator==(const KeyIterator& b) {return iter_==b.iter_;}
    bool operator!=(const KeyIterator& b) {return iter_!=b.iter_;}
};

int main() {
    std::map<std::string,int> m;
    KeyIterator<std::map<std::string,int> > ki;
    for(ki=m.begin(); ki!=m.end(); ++ki)
        cout << *ki;
}

http://codepad.org/4wxFGGNV
没有比那更轻量级。但是,它需要在 map 类型上模板化迭代器,而不是键类型,这意味着如果您试图隐藏内部,则必须提供一些实现细节。

答案 2 :(得分:0)

我想主要的问题是你只想要一个模板参数而不是两个:KeyIterator<string>而不是KeyIterator<string, X>

要实现这一点,您可能需要一种称为类型擦除的技术。但这将涉及动态分配和虚拟方法调用。

如果是后者,你可以只包装一个map<string, X>::const_iterator,make dereferencing返回对该键的引用,并提供带有map迭代器的构造函数。