字符串转整数atoi

字符串转整数很容易把代码写得难看。下面这段代码算是难得的优雅了,精妙的地方在于使用负数存储中间值,避免了过多的边界检查,而且将遍历单独抽了出来,虽然多了次遍历,但是代码因此变得整洁,是个划算的权衡。

 

    bool IsValid(const char *str, int len) {
        if (str == NULL || *str == '\0')
            return false;
        if (str[0] == '-' && (len == 1 || str[1] == '0'))
            return false;
        if (str[0] == '0' && len > 1)
            return false;
        for (int i = 1; i < len; ++i) {
            if (!isdigit(str[i]))
                return false;
        }
        return true;
    }

    int Atoi(const char *str, int len) {
        if (!IsValid(str, len))
            return 0;

        bool positive = str[0] == '-' ? false : true;
        int minq = INT_MIN / 10;
        int minr = INT_MIN % 10;

        int ret = 0;
        for (int i = positive ? 0 : 1; i < len; ++i) {
            int v = '0' - str[i];
            if (ret < minq || (ret == minq && v < minr)) // overflow
                return 0;
            ret = ret * 10 + v;
        }
        if (ret == INT_MIN && positive) // overflow
            return 0;

        return positive ? -ret : ret;
    }

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章