c++ - 从结构访问数据时，运算符重载在哪里？

Question

我在结构中有一个函数，可以对结构中的向量进行排序。但是要比较向量中的两个元素，我需要同一个结构中另一个变量的值。我想知道我应该在哪里保留运算符重载或比较函数才能使这种工作正常工作。我在下面的粘贴中给出了一个示例。

#include<vector>
#include<algorithm>

struct Square{
    int color; //value 1 to 10
};
struct State{
    vector<Square> list;
    int color_weight[] = {4,3,5,2,4,1,6,4,5,9}; //These values keep changing.
    bool operator<(Square& a, Square& b);
    void sortTheList();

};    

bool State::operator<(Square& a, Square& b){
    if (color_weight[a.color]< color_weight[b.color]){
        return true;
    }
    return false;
}

void Square::sortTheList(){
    sort(list.begin(),list.end());
}

这当然行不通。我已经为比较功能尝试了许多其他签名和范围，但似乎没有任何效果。

知道可以在这里做什么吗？

Answer 1

您将使用一个比较器来保持对它所需的额外状态的引用，而不是operator<. 像这样的东西：

struct CompareWeight {
    CompareWeight(int const * weight) : weight(weight) {}
    bool operator()(Square const & lhs, Square const & rhs) {
        return weight[lhs.color] < weight[rhs.color];
    }
    int const * weight;
};

void Square::sortTheList() {
    std::sort(list.begin(), list.end(), CompareWeight(color_weight));
}