partition.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSL-1.0
 
#ifndef TETL_ALGORITHM_PARTITION_HPP
#define TETL_ALGORITHM_PARTITION_HPP
 
#include <etl/_algorithm/find_if_not.hpp>
#include <etl/_algorithm/iter_swap.hpp>
#include <etl/_iterator/next.hpp>
 
namespace etl {

template <typename ForwardIt, typename Predicate>
constexpr auto partition(ForwardIt first, ForwardIt last, Predicate p) -> ForwardIt
{
    first = etl::find_if_not(first, last, p);
    if (first == last) {
        return first;
    }
 
    for (ForwardIt i = etl::next(first); i != last; ++i) {
        if (p(*i)) {
            etl::iter_swap(i, first);
            ++first;
        }
    }
    return first;
}
 
} // namespace etl
 
#endif // TETL_ALGORITHM_PARTITION_HPP