Question

我有地图，哪些键是std :: string。我想在地图中找到以"DUPA/"前缀开头的那些元素。找到下限很容易，但上限有点问题。我写了这样一段代码：

const char* prefix = "DUPA/";
const char* firstAfterPrefix = "DUPA0";
auto prefixedBeginIt = myMap.upper_bound(prefix);
auto prefixedEndIt = myMap.lower_bound(firstAfterPrefix);

代码工作正常，但我不认为它很优雅，因为必须知道0首先在ASCII表的/旁边。第二种方法是复制前缀和增加最后一个符号。你知道更优雅的解决方案吗？

Answer 1

我认为你提到的解决方案已经是最优雅的了。 KISS方式失去了很多性能，即每次检查密钥：

while(prefixedBeginIt->first == prefix)
{
 //...
 ++prefixedBeginIt;
}

因此，我认为计算下一个字符是最好的方法：

std::string firstAfterPrefix = prefix;
++firstAfterPrefix[firstAfterPrefix.length() - 1];
auto prefixedEndIt = myMap.lower_bound(firstAfterPrefix);

Answer 2

如果您认为CHAR_MAX不是字符串中的有效字符，那么您可以通过附加firstAfterPrefix（或者字符类型的最大值，如果它是）来创建CHAR_MAX不是char）。

std::string prefix = "DUPA/";

constexpr auto char_max = std::numeric_limits<decltype(prefix)::value_type>::max();
std::string firstAfterPrefix = prefix + char_max;

auto prefixedBeginIt = myMap.lower_bound(prefix);
auto prefixedEndIt = myMap.lower_bound(firstAfterPrefix);

请注意lower_bound对两个边界的使用。像Gill一样，我使用std::string来简化说明。

如果您可以使用C ++ 14并指定容器的Compare模板参数，那么另一种方法是使用自定义探测对象：

struct PrefixProbe { string_view prefix; };
bool operator<(PrefixProbe a, std::string_view b) { return a.prefix < b; }
bool operator<(std::string_view a, PrefixProbe b) { return a < b.prefix && a.substr(0, b.prefix.size()) == b.prefix; }

std::map<std::string, myValue, std::less<>> myMap;
//                             ^~~~~~~~~~~
//                             where the magic happens

auto prefixBegin = myMap.lower_bound(PrefixProbe { prefix });
auto prefixEnd = myMap.upper_bound(PrefixProbe { prefix });

string_view是C ++ 17，但不需要这样做。

equal_range会将最后两行减少为一行：

auto [ prefixBegin, prefixEnd ] = myMap.equal_range(PrefixProbe { prefix });

如果您准备使用STL算法而不是容器成员函数，那么这可以在不改变容器类型的情况下完成，但效率会降低：

auto prefixBegin = std::lower_bound(cbegin(myMap), cend(myMap), PrefixProbe { prefix }, std::less<>{});
auto prefixEnd = std::upper_bound(cbegin(myMap), cend(myMap), PrefixProbe { prefix }, std::less<>{});

auto [ prefixBegin, prefixEnd ] = std::equal_range(cbegin(myMap), cend(myMap), PrefixProbe { prefix }, std::less<>{});

找到std :: map中给定前缀或std :: set中元素的键的优雅方法

2 个答案: