有没有实现条件累加的boost或stl函数？

Question

我想实现将一个整数数组相加直到满足某个条件的代码。这是我的代码：

int x [5];

int GetIndex (int val)
{
    int i; 
    int xtotal=0; 

    for (i=0; i < 5; ++i) {
        if (xtotal > val) 
            break;
        xtotal += x [i];
    }


    return i;    
}

我正在使用 c++03。我遇到了 stl 累积功能，但我不想把所有东西都加起来。我想将值相加，直到它大于输入。数组只是一个例子，可以用向量或其他东西代替。

Answer 1

你可以自己写。 这是来自http://www.richelbilderbeek.nl/CppAccumulate_if.htm的那个

template
   <
   typename InputIterator,
   typename ElementType,
   typename Predicate
   >
 const ElementType accumulate_if(
   InputIterator first,
   const InputIterator last,
   ElementType init,
   const Predicate predicate)
 {
   for (; first != last; ++first)
     if (predicate(*first)) init += *first;
   return init;
 }

他还有一个可以进行二元运算的，以防你除了加法之外还想做一些事情。

Answer 2

你可以使用类似 (c++03):

template<class InputIt, class T>
std::pair<InputIt, T>
accumulate_until(InputIt begin, InputIt end, T value, T thresold_value)
{
    for (InputIt it = begin; it != end; ++it) {
        value += *it;
        if (thresold_value < value) {
            return std::make_pair(it, value);
        }
    }
    return std::make_pair(end, value);
}

或者更通用的解决方案（c++03）：

namespace detail
{
    template <typename T, typename Pred>
    class InternalPredicate
    {
    public:
        InternalPredicate(T& value, Pred& pred) : value(value), pred(pred) {}

        bool operator () (const T& t) {
            value += t;
            return pred(value);
        }
    private:
        T& value;
        Pred& pred;
    };
}

template<class InputIt, class T, class Pred>
std::pair<InputIt, T>
accumulate_until(InputIt begin, InputIt end, T value, Pred pred)
{
    InputIt it = std::find_if(begin, end, detail::InternalPredicate<T, Pred>(value, pred));
    return std::make_pair(it, value);
}

或在 C++11 中：

template<class InputIt, class T, class Pred>
std::pair<InputIt, value>
accumulate_until(InputIt begin, InputIt end, T value, Pred pred)
{
    auto internal_pred = [&value, &pred] (const T& t) {
        value += t;
        return pred(value);
    };
    InputIt it = std::find_if(begin, end, internal_pred);
    return std::make_pair(it, value);
}