如何基于STL/BOOST等库实现每N个元素取最大值并返回索引？

当然有办法实现这个需求！你可以通过自定义遍历逻辑或者借助第三方库的迭代器适配器，轻松实现「每隔N个元素比较并找出最大值索引」的功能，不管是用STL原生工具还是Boost库都能搞定。

既然你提到了要修改std::max_element的循环步长，那我们可以直接参考标准库的实现逻辑，自己写一个支持指定步长的版本。这个方法不需要依赖第三方库，完全基于STL就能实现。

下面是完整的模板函数实现，既可以返回最大值的迭代器，也可以直接返回其索引（针对随机访问迭代器）：

#ifndef NEWTON_ALGORITHM_SEARCH_MAX_INDEX_HPP
#define NEWTON_ALGORITHM_SEARCH_MAX_INDEX_HPP
#include <iterator>
#include <functional> // 用std::less作为默认比较器

namespace newton {

// 核心函数：返回每隔stride步的元素中的最大值迭代器
template <typename ForwardIterator, typename Compare = std::less<typename std::iterator_traits<ForwardIterator>::value_type>>
ForwardIterator max_element_strided(ForwardIterator first, ForwardIterator last, size_t stride, Compare comp = Compare()) {
    // 处理空容器的边界情况
    if (first == last) {
        return last;
    }

    ForwardIterator result = first;
    // 跳过第一个元素，从下一个步长位置开始遍历
    first += stride;
    // 循环步长改为stride，而非默认的++
    for (; first < last; first += stride) {
        if (comp(*result, *first)) {
            result = first;
        }
    }
    return result;
}

// 重载版本：直接返回最大值的索引（仅支持随机访问迭代器，如vector、array）
template <typename RandomAccessIterator, typename Compare = std::less<typename std::iterator_traits<RandomAccessIterator>::value_type>>
typename std::iterator_traits<RandomAccessIterator>::difference_type max_index_strided(RandomAccessIterator first, RandomAccessIterator last, size_t stride, Compare comp = Compare()) {
    auto max_iter = max_element_strided(first, last, stride, comp);
    return std::distance(first, max_iter);
}

} // namespace newton

#endif // NEWTON_ALGORITHM_SEARCH_MAX_INDEX_HPP

代码说明： #

• 模板参数支持任意前向迭代器，兼容大部分STL容器（比如std::list、std::vector）
• 默认使用std::less比较器，你也可以传入自定义比较函数（比如找最小值就传std::greater）
• 针对随机访问迭代器提供了索引返回的重载，使用std::distance计算迭代器对应的位置
• 处理了空容器的边界情况，避免非法访问

如果你愿意使用Boost库，那么boost::strided_iterator可以帮你省去手写循环的麻烦——它能把普通迭代器包装成指定步长的迭代器，直接传给std::max_element就能实现需求。

示例代码如下：

#include <vector>
#include <algorithm>
#include <boost/iterator/strided_iterator.hpp>

int main() {
    std::vector<int> nums = {10, 2, 8, 5, 15, 3, 7, 12, 6};
    const size_t stride = 3; // 每隔3个元素，即检查索引0、3、6的元素

    // 用Boost包装出步长为stride的迭代器
    auto strided_begin = boost::make_strided_iterator(nums.begin(), stride);
    auto strided_end = boost::make_strided_iterator(nums.end(), stride);

    // 直接调用标准库的std::max_element
    auto max_strided_iter = std::max_element(strided_begin, strided_end);

    // 转换回原容器的迭代器，再计算索引
    auto original_iter = max_strided_iter.base();
    size_t max_index = std::distance(nums.begin(), original_iter);

    // 输出结果：这里max_index会是4（对应元素15）
    return 0;
}

优势： #

• 完全复用标准库的std::max_element，不需要自己实现遍历逻辑
• 代码更简洁，可读性更高
• Boost的迭代器适配器支持多种迭代器类型，兼容性好

• 如果使用的是非随机访问迭代器（比如std::list），std::distance的时间复杂度是O(n)，此时更推荐直接使用返回迭代器的版本
• 要确保步长stride大于0，否则会陷入无限循环
• 自定义比较函数时，要保证其满足严格弱序的要求，避免出现未定义行为

内容的提问来源于stack exchange，提问作者Moises Rojo