c++ - 如何创建一个使用局部变量对元素进行排序的集合？

Question

如何创建一个集合，其中元素始终使用某个局部变量进行排序？

这是我正在尝试做的一个简化示例。

int x[5] {9, 2, 3, 1, 8};
set<int, ???> my_set;
my_set.insert(0);
my_set.insert(1);
my_set.insert(4);
for (int a : my_set)
    cout << a << " ";   // I want the answer 1 4 0 because x[1] < x[4] < x[0]

我想我可以使用 struct 来做到这一点，但我不确定 x 的使用如何改变事情。

Answer 1

你可以用 lambda 像这样设置它

int x[5] {9, 2, 3, 1, 8};
auto comparator = [&](int a, int b){ return x[a] < x[b]; };
std::set<int, decltype(comparator)> my_set(comparator);