std:: partition_point

检查已分区的范围 [ first , last ) 并定位第一个分区的结束位置，即首个不满足 p 的元素，若所有元素均满足 p 则返回 last 。

如果范围 [ first , last ) 中的元素 elem 未按表达式 bool ( p ( elem ) ) 进行 分区 ，则行为未定义。

参数

返回值

指向第一个分区末尾的迭代器，位于 [ first , last ) 范围内，若所有元素均满足 p 则返回 last 。

复杂度

给定 \(\scriptsize N\) N 为 std:: distance ( first, last ) ，执行 \(\scriptsize O(log(N))\) O(log(N)) 次谓词 p 的应用。

注释

该算法是 std::lower_bound 的更通用形式，可通过 std::partition_point 配合谓词 [ & ] ( const auto & e ) { return e < value ; } ) ; 来表达。

可能的实现

template<class ForwardIt, class UnaryPred>
constexpr //< since C++20
ForwardIt partition_point(ForwardIt first, ForwardIt last, UnaryPred p)
{
    for (auto length = std::distance(first, last); 0 < length; )
    {
        auto half = length / 2;
        auto middle = std::next(first, half);
        if (p(*middle))
        {
            first = std::next(middle);
            length -= (half + 1);
        }
        else
            length = half;
    }
    return first;
}

示例

#include <algorithm>
#include <array>
#include <iostream>
#include <iterator>
auto print_seq = [](auto rem, auto first, auto last)
{
    for (std::cout << rem; first != last; std::cout << *first++ << ' ') {}
    std::cout << '\n';
};
int main()
{
    std::array v{1, 2, 3, 4, 5, 6, 7, 8, 9};
    auto is_even = [](int i) { return i % 2 == 0; };
    std::partition(v.begin(), v.end(), is_even);
    print_seq("After partitioning, v: ", v.cbegin(), v.cend());
    const auto pp = std::partition_point(v.cbegin(), v.cend(), is_even);
    const auto i = std::distance(v.cbegin(), pp);
    std::cout << "Partition point is at " << i << "; v[" << i << "] = " << *pp << '\n';
    print_seq("First partition (all even elements): ", v.cbegin(), pp);
    print_seq("Second partition (all odd elements): ", pp, v.cend());
}

After partitioning, v: 8 2 6 4 5 3 7 1 9
Partition point is at 4; v[4] = 5
First partition (all even elements): 8 2 6 4
Second partition (all odd elements): 5 3 7 1 9

参见