Bagaimana shift_right () dimaksudkan untuk diimplementasikan dalam C ++ 20?

9

Di C ++ 20, <algorithm>header mendapatkan dua algoritma baru: shift_left()danshift_right() . Keduanya menerima LegacyForwardIterator. Sebab shift_left(), ditentukan bahwa "gerakan dilakukan dengan urutan imulai dari ​0"; untuk shift_right(), ditentukan bahwa "jika ForwardItmemenuhi persyaratan LegacyBidirectionalIterator, maka gerakan dilakukan dalam urutan menurun imulai dari last - first - n - 1".

Saya bisa memikirkan cara yang cukup mudah untuk diterapkan 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);
}

Jika ForwardItmemenuhi persyaratan LegacyBidirectionalIterator, saya dapat melihat bahwa shift_right()dapat diimplementasikan dengan cara yang sangat mirip shift_left(). Namun, tidak jelas sama sekali bagaimana seseorang dapat menerapkan shift_right()untuk iterator forward non-bidirectional.

Saya telah menemukan sebuah algoritma yang menggunakan ruang di [first, first+n)sebagai ruang awal untuk bertukar elemen, tetapi tampaknya sedikit lebih boros daripada algoritma untuk di shift_left()atas:

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;
}

Apakah akan ada cara implementasi yang lebih baik atau "dimaksudkan" shift_right()?

Bernard
sumber
Implementasi lebih suka menggunakan std::movedaripada std::copy...
Aconcagua
@Aconcagua Ups, ya, saya akan mengedit pertanyaan.
Bernard

Jawaban:

6

Ini adalah contoh implementasi untuk shift: https://github.com/danra/shift_proposal/blob/master/shift_proposal.h

Dari dokumen proposal: 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
sumber
3
Saya heran mengapa void(++trail)...
YSC
@YSC melindungi dari peringatan "hasil terbuang" yang terlalu bersemangat
Caleth
@ll Itulah yang saya pikirkan.
YSC
5
@YSC Mungkin untuk melindungi terhadap operator koma kelebihan beban yang tidak seharusnya dipanggil.
kenari
@walnut ho kamu mungkin benar!
YSC