fix implementation: search_n

This commit is contained in:
bolero-MURAKAMI 2012-12-13 18:08:59 +09:00
parent 9b0d1a7933
commit d6914ddd72
3 changed files with 124 additions and 21 deletions

View file

@ -22,6 +22,7 @@
#include "./equal.cpp"
#include "./is_permutation.cpp"
#include "./search.cpp"
#include "./search_n.cpp"
#ifdef TESTSPR_CPP_INCLUDE_DISABLE_SPROUT_LIBS_ALGORITHM_TEST_NON_MODIFYIING_CPP
# undef TESTSPR_CPP_INCLUDE
@ -45,6 +46,7 @@ namespace testspr {
testspr::algorithm_equal_test();
testspr::algorithm_is_permutation_test();
testspr::algorithm_search_test();
testspr::algorithm_search_n_test();
}
} // namespace testspr

View file

@ -0,0 +1,101 @@
#ifndef SPROUT_LIBS_ALGORITHM_TEST_SEARCH_N_CPP
#define SPROUT_LIBS_ALGORITHM_TEST_SEARCH_N_CPP
#include <sprout/algorithm/search_n.hpp>
#include <sprout/array.hpp>
#include <sprout/container.hpp>
#include <testspr/tools.hpp>
namespace testspr {
static void algorithm_search_n_test() {
using namespace sprout;
{
SPROUT_STATIC_CONSTEXPR auto arr1 = array<int, 10>{{1, 2, 3, 5, 5, 5, 5, 8, 9, 10}};
{
SPROUT_STATIC_CONSTEXPR auto found = sprout::search_n(
sprout::begin(arr1),
sprout::end(arr1),
2,
5
);
TESTSPR_BOTH_ASSERT(found == sprout::begin(arr1) + 3);
}
{
SPROUT_STATIC_CONSTEXPR auto found = sprout::search_n(
sprout::begin(arr1),
sprout::end(arr1),
4,
5
);
TESTSPR_BOTH_ASSERT(found == sprout::begin(arr1) + 3);
}
{
SPROUT_STATIC_CONSTEXPR auto found = sprout::search_n(
sprout::begin(arr1),
sprout::begin(arr1) + 5,
2,
5
);
TESTSPR_BOTH_ASSERT(found == sprout::begin(arr1) + 3);
}
{
SPROUT_STATIC_CONSTEXPR auto found = sprout::search_n(
sprout::begin(arr1),
sprout::begin(arr1) + 5,
4,
5
);
TESTSPR_BOTH_ASSERT(found == sprout::begin(arr1) + 5);
}
{
SPROUT_STATIC_CONSTEXPR auto found = sprout::search_n(
sprout::begin(arr1),
sprout::end(arr1),
2,
5,
testspr::equal_to<int>()
);
TESTSPR_BOTH_ASSERT(found == sprout::begin(arr1) + 3);
}
{
SPROUT_STATIC_CONSTEXPR auto found = sprout::search_n(
sprout::begin(arr1),
sprout::end(arr1),
4,
5,
testspr::equal_to<int>()
);
TESTSPR_BOTH_ASSERT(found == sprout::begin(arr1) + 3);
}
{
SPROUT_STATIC_CONSTEXPR auto found = sprout::search_n(
sprout::begin(arr1),
sprout::begin(arr1) + 5,
2,
5,
testspr::equal_to<int>()
);
TESTSPR_BOTH_ASSERT(found == sprout::begin(arr1) + 3);
}
{
SPROUT_STATIC_CONSTEXPR auto found = sprout::search_n(
sprout::begin(arr1),
sprout::begin(arr1) + 5,
4,
5,
testspr::equal_to<int>()
);
TESTSPR_BOTH_ASSERT(found == sprout::begin(arr1) + 5);
}
}
}
} // namespace testspr
#ifndef TESTSPR_CPP_INCLUDE
# define TESTSPR_TEST_FUNCTION testspr::algorithm_search_n_test
# include <testspr/include_main.hpp>
#endif
#endif // #ifndef SPROUT_LIBS_ALGORITHM_TEST_SEARCH_N_CPP

View file

@ -3,33 +3,33 @@
#include <sprout/config.hpp>
#include <sprout/iterator/operation.hpp>
#include <sprout/iterator/value_iterator.hpp>
#include <sprout/functional/equal_to.hpp>
#include <sprout/algorithm/search.hpp>
namespace sprout {
// Copyright (C) 2011 RiSK (sscrisk)
// 25.2.13 Search
template<typename ForwardIterator, typename Size, typename T>
inline SPROUT_CONSTEXPR ForwardIterator
search_n(ForwardIterator first, ForwardIterator last, Size count, T const& value) {
return first == last || count == 0 ? first
: sprout::next(first) == last && count > 1 ? last
: *first == value
&& sprout::search_n(sprout::next(first), last, count - 1, value) == sprout::next(first)
? first
: sprout::search_n(sprout::next(first), last, count, value)
;
}
//
// recursion depth:
// [first, last) is RandomAccessIterator -> O(log N)
// otherwise -> O(N)
//
template<typename ForwardIterator, typename Size, typename T, typename BinaryPredicate>
inline SPROUT_CONSTEXPR ForwardIterator
search_n(ForwardIterator first, ForwardIterator last, Size count, T const& value, BinaryPredicate pred) {
return first == last || count == 0 ? first
: sprout::next(first) == last && count > 1 ? last
: *first == value
&& sprout::search_n(sprout::next(first), last, count - 1, value, pred) == sprout::next(first)
? first
: sprout::search_n(sprout::next(first), last, count, value, pred)
;
typedef sprout::value_iterator<T const&> iterator;
typedef typename iterator::difference_type difference_type;
return sprout::search(
first, last,
iterator(value, static_cast<difference_type>(count)), iterator(value, 0),
pred
);
}
template<typename ForwardIterator, typename Size, typename T>
inline SPROUT_CONSTEXPR ForwardIterator
search_n(ForwardIterator first, ForwardIterator last, Size count, T const& value) {
return sprout::search_n(first, last, count, value, sprout::equal_to<>());
}
} // namespace sprout