如何在 C++ Map 中使用自定义结构

Question

我想用一个C++的map结构，比如map<vector<DFSCode>, vector<PDB>> candidate，DFSCode和PDB是我定义的两个结构。

class DFS {
public:
    int from;
    int to;
    int fromlabel;
    int elabel;
    int tolabel;
    DFS(): from(0), to(0), fromlabel(0), elabel(0), tolabel(0) {};
};

struct DFSCode: public vector <DFS> {
public:
    void push (int from, int to, int fromlabel, int elabel, int tolabel)
    {
        resize (size() + 1);
        DFS &d = (*this)[size()-1];

        d.from = from;
        d.to = to;
        d.fromlabel = fromlabel;
        d.elabel = elabel;
        d.tolabel = tolabel;
    }
    void pop () { resize (size()-1); }
};

class PDB {
public:
    unsigned int tid;
    unsigned int gid;
    void push(int did, int vid, int vlabel)
    {
        tuple[did].vid = vid;
        tuple[did].vlabel = vlabel;
    }
    PDB(): tid(0), gid(0), tuple(0) {};
};

我会生成很多包含vector<DFSCode>和PDB的数据，因为一个vector<DFSCode>可能有很多PDB，我想用vector<PDB>来存储它们。 我想做的是：

vector<DFSCode> tempdfscodeList;
PDB             temppdb;
map<vector<DFSCode>, vector<PDB>> candidate;
for each `vector<DFSCode>` and `PDB` pair I generate
    candidate[tempdfscodeList].push_back(temppdb);

第一个问题是：上面的代码是否满足我对“一个vector<DFSCode>包含许多PDB”的期望？

第二个问题是：我知道我必须实现一个类似的map方法，因为我使用vector<DFSCode>作为我的key，但我不知道如何实现。我试着写一个。但似乎不能满足我对“一个vector<DFSCode> 包含许多PDB”的期望，有人可以帮助我吗？ :)

class dfscodeListCompare {  // compare vector<DFSCode>
public:
    bool operator() (const vector<DFSCode> &c1, const vector<DFSCode> &c2) const
    {
        for(int I = 0; I < c1.size(); I++) {
            if(c1[I].size() == c2[I].size()) {  // the size must be the same
                for(int j = 0; j < c1[I].size(); j++) {
                    if((c1[I][j].from != c2[I][j].from) || (c1[I][j].to != c2[I][j].to) || (c1[I][j].fromlabel != c2[I][j].fromlabel) || (c1[I][j].elabel != c2[I][j].elabel) || (c1[I][j].tolabel != c2[I][j].tolabel))
                        return false;   // if there exist one different
                }
            }
            else
                return false;
        }
        return true;    // pass all condition
    }
};

Answer 1

DFSCode 的向量可以包含多个DFSCode。由于DFSCode 可以 包含很多DFS，DFSCode 的向量可以包含很多很多DFS。

关于您的代码：一些建议：

• 使用 `push_back` 和 `pop_back`，而不是 `resize`。更多 惯用语。你的函数 `push` 应该开始： push_back(DFS()); back().from() = 从； ...
• 给 `DFS` 一个构造函数，它接受它需要的参数： DFS::DFS(int from, int to, int fromLabel, int eLabel, int toLabel) : 从( 从 ) , 到 ( 到 ) , 来自标签（来自标签） , 电子标签（电子标签） , toLabel( toLabel ) { } 然后 `push` 变得很简单： push_back（DFS（从，到，从标签，电子标签，到标签））；
• 不要从 `std::vector` 继承。使其成为数据成员。

关于您关于订购功能的问题， std::vector<DFSCode> 基本上是一个二维结构。这 可以通过lexicographical_compare优雅地处理：

struct CompareDFSCode
{
    bool operator()( DFS const& lhs, DFS const& rhs ) const
    {
        if ( lhs.from != rhs.from )
            return lhs.from < rhs.from;
        else if ( lhs.to != rhs.to )
            return lhs.to < rhs.to;
        else if ( lhs.fromLabel != rhs.fromLabel )
            return lhs.fromLabel < rhs.fromLabel;
        else if ( lhs.eLabel != rhs.eLabel )
            return lhs.eLabel < rhs.eLabel;
        else
            return lhs.toLabel < rhs.toLabel;
    }

    bool operator()( DFSCode const& lhs, DFSCode const& rhs ) const
    {
        return std::lexicographical_compare(
            lhs,begin(), lhs.end(),
            rhs.begin(), rhs.end(),
            *this );
    }

    bool operator()(
            std::vector<DFSCode> const& lhs,
            std::vector<DFSCode> const& rhs ) const
    {
        return std::lexicographical_compare(
            lhs.begin(), lhs.end(),
            rhs.begin(), rhs.end(),
            *this );
    }
};

编辑：

有一点我忘了提。有了上面的对比 运算符，向量中项目的顺序很重要。如果这是 不可接受，那么您可能最终必须对元素进行排序 首先（临时）。

Answer 2

第一个问题是：上面的代码是否满足我对“一个向量包含多个PDB”的期望？

使用multimap 而不是map。 map 只能有一个键值。它类似于 一对一 关系。 multimap 一个键可以有多个值。

你不需要像 antonio 建议的那样继承 DFScode。