如何正确实现->和(*)。因此它们的行为类似于->和(*)。在迭代器中

时间:2019-02-07 08:39:12

标签: c++ c++11 operator-overloading dereference

我想为定制容器的迭代器实现*和->运算符。我的代码无法编译。在“最小非工作示例”下面,该示例显示了它如何与std :: map一起工作,但在我的代码中没有。

#include <vector>
#include <map>
#include <iostream>

struct thing {
  float f[1];
  typedef std::pair<int,float> key_data;

  struct iterator {
    int pos;
    const float *f;
    key_data operator*() const { return key_data(pos,f[pos]); }
    key_data *operator->() const { return &key_data(pos,f[pos]); }
  };

  iterator begin() const { return {.f = f, .pos = 0}; }
};

template<typename T> void test(T iter) {
  (*iter).second = 1.0; std::cout << (*iter).second;

  iter->second = 2.0; std::cout << iter->second;
}

int main() {
  std::map<int,float> fmap; fmap[0] = 0.0;
  test(fmap.begin());
  std::cout << fmap[0];

  thing f;
  test(f.begin());
  std::cout << f.f[0];
}

我想编译它:),并打印122122。编译时的错误消息是:

access.cc:13:43: error: taking the address of a temporary object of type 'key_data'
      (aka 'pair<int, float>') [-Waddress-of-temporary]
    key_data *operator->() const { return &key_data(pos,f[pos]); }

access.cc:20:18: error: expression is not assignable
  (*iter).second = 1.0; std::cout << (*iter).second;

第一个:足够公平,std :: pair <>创建一个不能通过引用返回的临时值;但是标准库如何做才能允许通常的->语法?

第二个:可能我还是想尝试将其分配给一个临时文件,但我无法猜出正确的语法是什么。

1 个答案:

答案 0 :(得分:2)

简单的是在迭代器中直接具有正确的类型:

struct thing {
  float f[1];
  using key_data = std::pair<int,float&>;

  struct iterator {
    key_data data;
    const key_data& operator*() { return data; }
    key_data *operator->() { return &data; }
  };

  iterator begin() { return {{0, f[0] }}; }
};

Demo

另一种解决方案是使用包装器,例如:

struct pair_wrapper
{
    int &first;
    float& second;

    pair_wrapper* operator->() { return this; }
};

struct thing {
  float f[1];
  typedef std::pair<int,float> key_data;

  struct iterator {
    int pos;
    float *f;
    pair_wrapper operator*() { return pair_wrapper{pos,f[pos]}; }
    pair_wrapper operator->() { return pair_wrapper{pos,f[pos]}; }
  };

  iterator begin() { return {0, f}; }
};

Demo

它使用operator->的“魔术”链接。