在一个封闭的范围内循环

时间:2010-03-16 19:19:09

标签: c++ overflow numeric-limits

您如何修复此代码?

template <typename T> void closed_range(T begin, T end)
{
    for (T i = begin; i <= end; ++i) {
        // do something
    }
}
  • T被约束为整数类型,可以是更广泛的类型,可以是有符号或无符号的

  • begin可以是numeric_limits<T>::min()

  • end可以是numeric_limits<T>::max()(在这种情况下,++i会溢出上述代码)

我有几种方式,但我不喜欢。

5 个答案:

答案 0 :(得分:6)

可能,

template <typename T> void closed_range(T begin, const T end)
    if (begin <= end) {
        do {
            // do something
        } while (begin != end && (++begin, true));
    }
}
诅咒,我的第一次尝试是错误的,上面的解决方案并不像我希望的那样漂亮。怎么样:

template <typename T> bool advance(T &value) { ++value; return true; }

template <typename T> void closed_range(T first, const T last)
    if (first <= last) {
        do {
            // do something
        } while (first != last && advance(first));
    }
}

即使T不是整数类型,std::advance也没有歧义,因为std::advance需要2个参数。因此,模板也可以使用例如随机访问迭代器,如果由于某种原因你想要一个封闭的范围。

或者一些集理论怎么样?显然,如果你只是在一个封闭的范围内写一个循环,那就太大了,但是如果你想要做很多事情,那么它就会使循环代码正确。不确定效率:在一个非常紧凑的循环中,您可能需要确保暂停对endof的呼叫:

#include <limits>
#include <iostream>

template <typename T>
struct omega {
    T val;
    bool isInfinite;
    operator T() { return val; }
    explicit omega(const T &v) : val(v), isInfinite(false) { }
    omega &operator++() {
        (val == std::numeric_limits<T>::max()) ? isInfinite = true : ++val;
        return *this;
    }
};

template <typename T>
bool operator==(const omega<T> &lhs, const omega<T> &rhs) {
    if (lhs.isInfinite) return rhs.isInfinite;
    return (!rhs.isInfinite) && lhs.val == rhs.val;
}
template <typename T>
bool operator!=(const omega<T> &lhs, const omega<T> &rhs) {
    return !(lhs == rhs);
}

template <typename T>
omega<T> endof(T val) { 
    omega<T> e(val);
    return ++e;
}

template <typename T>
void closed_range(T first, T last) {
    for (omega<T> i(first); i != endof(last); ++i) {
        // do something
        std::cout << i << "\n";
    }
}

int main() {
    closed_range((short)32765, std::numeric_limits<short>::max());
    closed_range((unsigned short)65533, std::numeric_limits<unsigned short>::max());
    closed_range(1, 0);
}

输出:

32765
32766
32767
65533
65534
65535

omega<T>对象上使用其他运算符时要小心。我只实现了演示的绝对最小值,并且omega<T>隐式转换为T,因此你会发现你可以编写表达式,这可能会抛弃omega对象的“无限”。你可以通过声明(不一定定义)一整套算术运算符来解决这个问题;或者如果isInfinite为真,则在转换中抛出异常;或者只是不要担心它,因为你不能意外地将结果转换回欧米茄,因为构造函数是明确的。但例如,omega<int>(2) < endof(2)为真,但omega<int>(INT_MAX) < endof(INT_MAX)为假。

答案 1 :(得分:5)

我的看法:

// Make sure we have at least one iteration
if (begin <= end)
{
    for (T i = begin; ; ++i)
    {
        // do something

        // Check at the end *before* incrementing so this won't
        // be affected by overflow
        if (i == end)
            break;
    }
}

答案 2 :(得分:4)

这很有效,而且相当清楚:

T i = begin;
do {
   ...
}
while (i++ < end);

如果你想了解begin >= end的特殊情况,你需要在Steve Jessop的解决方案中添加另一个if

答案 3 :(得分:3)

template <typename T, typename F>
void closed_range(T begin, T end, F functionToPerform) 
{
    for (T i = begin; i != end; ++i) { 
        functionToPerform(i);
    }
    functionToPerform(end);
} 

答案 4 :(得分:0)

编辑:重新设计的东西与OP更接近。

#include <iostream>
using namespace std;

template<typename T> void closed_range(T begin, T end)
{
    for( bool cont = (begin <= end); cont; )
    {
        // do something
        cout << begin << ", ";
        if( begin == end )
            cont = false;
        else
            ++begin;
    }

    // test - this should return the last element
    cout << " --  " << begin;
}
int main()
{
    closed_range(10, 20);
    return 0;
}

输出结果为:

  

10,11,12,13,14,15,16,17,18,   19,20, - 20