W jaki sposób shift_right () ma być zaimplementowany w C ++ 20?

9

W C ++ 20 <algorithm>nagłówek zyskuje dwa nowe algorytmy: shift_left()ishift_right() . Oba akceptują dowolny LegacyForwardIterator. W przypadku shift_left(), podano, że „ruchy są wykonywane w rosnącej kolejności ipoczynając od ​0”; dla shift_right(), określono, że „jeżeli ForwardItspełnia wymagania LegacyBidirectionalIterator, a następnie przemieszcza wykonywane są w kolejności malejącej ipocząwszy od last - first - n - 1”.

Mogę wymyślić dość łatwy sposób wdrożenia shift_left():

template <typename ForwardIt>
constexpr inline ForwardIt shift_left(ForwardIt first, ForwardIt last, typename std::iterator_traits<ForwardIt>::difference_type n) {
    if (n <= 0) return last;
    ForwardIt it = first;
    for (; n > 0; --n, ++it) {
        if (it == last) return first;
    }
    return std::move(it, last, first);
}

Jeśli ForwardItspełnia wymagania LegacyBidirectionalIterator, widzę, że shift_right()można je zaimplementować w bardzo podobny sposób shift_left(). Jednak nie jest wcale jasne, jak można zaimplementować shift_right()dla dwukierunkowego iteratora do przodu.

Wymyśliłem algorytm, który wykorzystuje tę przestrzeń [first, first+n)jako przestrzeń do rysowania do zamiany elementów, ale wydaje się nieco bardziej marnotrawny niż algorytm dla shift_left()powyższego:

template <typename ForwardIt>
constexpr inline ForwardIt shift_right(ForwardIt first, ForwardIt last, typename std::iterator_traits<ForwardIt>::difference_type n) {
    if (n <= 0) return first;
    ForwardIt it = first;
    for (; n > 0; --n, ++it) {
        if (it == last) return last;
    }
    ForwardIt ret = it;
    ForwardIt ret_it = first;
    for (; it != last; ++it) {
        std::iter_swap(ret_it, it);
        ret_it++;
        if (ret_it == ret) ret_it = first;
    }
    return ret;
}

Czy byłby lepszy lub „zamierzony” sposób wdrożenia shift_right()?

Bernard
źródło
Wdrażanie wolałoby raczej std::movezamiast std::copy...
Aconcagua,
@Aconcagua Ups, tak, zmienię pytanie.
Bernard

Odpowiedzi:

6

To jest przykładowa implementacja zmian: https://github.com/danra/shift_proposal/blob/master/shift_proposal.h

Z dokumentu wniosku: http://www.open-std.org/jtc1/sc22/wg21/docs/papers/2017/p0769r0.pdf

#include <algorithm>
#include <iterator>
#include <type_traits>
#include <utility>

template<class I>
using difference_type_t = typename std::iterator_traits<I>::difference_type;

template<class I>
using iterator_category_t = typename std::iterator_traits<I>::iterator_category;

template<class I, class Tag, class = void>
constexpr bool is_category = false;
template<class I, class Tag>
constexpr bool is_category<I, Tag, std::enable_if_t<
    std::is_convertible_v<iterator_category_t<I>, Tag>>> = true;

/// Increment (decrement for negative n) i |n| times or until i == bound,
/// whichever comes first. Returns n - the difference between i's final position
/// and its initial position. (Note: "advance" has overloads with this behavior
/// in the Ranges TS.)
template<class I>
constexpr difference_type_t<I> bounded_advance(
    I& i, difference_type_t<I> n, I const bound)
{
    if constexpr (is_category<I, std::bidirectional_iterator_tag>) {
        for (; n < 0 && i != bound; ++n, void(--i)) {
            ;
        }
    }

    for(; n > 0 && i != bound; --n, void(++i)) {
        ;
    }

    return n;
}

template<class ForwardIt>
ForwardIt shift_left(ForwardIt first, ForwardIt last, difference_type_t<ForwardIt> n)
{
    if (n <= 0) {
        return last;
    }

    auto mid = first;
    if (::bounded_advance(mid, n, last)) {
        return first;
    }

    return std::move(std::move(mid), std::move(last), std::move(first));
}

template<class ForwardIt>
ForwardIt shift_right(ForwardIt first, ForwardIt last, difference_type_t<ForwardIt> n)
{
    if (n <= 0) {
        return first;
    }

    if constexpr (is_category<ForwardIt, std::bidirectional_iterator_tag>) {
        auto mid = last;
        if (::bounded_advance(mid, -n, first)) {
            return last;
        }
        return std::move_backward(std::move(first), std::move(mid), std::move(last));
    } else {
        auto result = first;
        if (::bounded_advance(result, n, last)) {
            return last;
        }

        // Invariant: next(first, n) == result
        // Invariant: next(trail, n) == lead

        auto lead = result;
        auto trail = first;

        for (; trail != result; ++lead, void(++trail)) {
            if (lead == last) {
                // The range looks like:
                //
                //   |-- (n - k) elements --|-- k elements --|-- (n - k) elements --|
                //   ^-first          trail-^                ^-result          last-^
                //
                // Note that distance(first, trail) == distance(result, last)
                std::move(std::move(first), std::move(trail), std::move(result));
                return result;
            }
        }

        for (;;) {
            for (auto mid = first; mid != result; ++lead, void(++trail), ++mid) {
                if (lead == last) {
                    // The range looks like:
                    //
                    //   |-- (n - k) elements --|-- k elements --|-- ... --|-- n elements --|
                    //   ^-first            mid-^         result-^         ^-trail     last-^
                    //
                    trail = std::move(mid, result, std::move(trail));
                    std::move(std::move(first), std::move(mid), std::move(trail));
                    return result;
                }
                std::iter_swap(mid, trail);
            }
        }
    }
}
vll
źródło
3
Zastanawiam się, dlaczego void(++trail)...
YSC
@YSC chroni przed nadgorliwymi ostrzeżeniami o „odrzuceniu wyniku”
Caleth
@vll Tak właśnie myślałem.
YSC
5
@YSC Prawdopodobnie w celu ochrony przed przeciążonymi operatorami przecinków, których nie należy wywoływać.
orzech
@walnut ho możesz mieć rację!
YSC