凌云的博客

行胜于言

LeetCode 算法题 8. 字符串转换整数 (atoi)

分类:algorithm| 发布时间:2016-06-27 23:58:00


题目

请你来实现一个 atoi 函数,使其能将字符串转换成整数。

首先,该函数会根据需要丢弃无用的开头空格字符,直到寻找到第一个非空格的字符为止。

当我们寻找到的第一个非空字符为正或者负号时,则将该符号与之后面尽可能多的连续数字组合起来,作为该整数的正负号;假如第一个非空字符是数字,则直接将其与之后连续的数字字符组合起来,形成整数。

该字符串除了有效的整数部分之后也可能会存在多余的字符,这些字符可以被忽略,它们对于函数不应该造成影响。

注意:假如该字符串中的第一个非空格字符不是一个有效整数字符、字符串为空或字符串仅包含空白字符时,则你的函数不需要进行转换。

在任何情况下,若函数不能进行有效的转换时,请返回 0。

说明:

假设我们的环境只能存储 32 位大小的有符号整数,那么其数值范围为 [−231,  231 − 1]。如果数值超过这个范围,请返回  INT_MAX (231 − 1) 或 INT_MIN (−231) 。

示例 1:

输入: "42"
输出: 42

示例 2:

输入: "   -42"
输出: -42

解释: 第一个非空白字符为 '-', 它是一个负号。   我们尽可能将负号与后面所有连续出现的数字组合起来,最后得到 -42 。

示例 3:

输入: "4193 with words"
输出: 4193

解释: 转换截止于数字 '3' ,因为它的下一个字符不为数字。

示例 4:

输入: "words and 987"
输出: 0

解释: 第一个非空字符是 'w', 但它不是数字或正、负号。 因此无法执行有效的转换。

示例 5:

输入: "-91283472332"
输出: -2147483648

解释: 数字 "-91283472332" 超过 32 位有符号整数范围。   因此返回 INT_MIN (−231) 。

解法

这道题也相当简单,就是需要注意要跳过前面的空格,以及处理数字开头的 '+' 以及 '-' 号。 同时还要处理结果溢出是返回类型的最大值或最小值。

class Solution {
public:
    int myAtoi(string str) {
        int sign = 1;
        int end = str.size() - 1, beg = 0;
        // skip space
        for (; beg <= end; ++beg) {
            if (str[beg] != ' ') {
                break;
            }
        }

        if (str[beg] == '-') {
            sign = -1;
            beg += 1;
        } else if (str[beg] == '+') {
            sign = 1;
            beg += 1;
        }

        long ret = 0;
        char c;
        for (; end >= beg; ++beg) {
            c = str[beg];
            if (c < '0' || c > '9') {
                return ret * sign;
            }

            ret *= 10;
            ret += (c - '0');
            if (ret * sign > INT_MAX) {
                return INT_MAX;
            } else if (ret * sign < INT_MIN) {
                return INT_MIN;
            }
        }
        return ret * sign;
    }
};