leetcode 8. 字符串转换整数 (atoi)

8. 字符串转换整数 (atoi)

问题描述

请你来实现一个 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) {
        string::iterator ite;
        int mul = 1;
        long ans = 0;
        for(ite = str.begin(); ite != str.end(); ite++)
        {
            if(*ite == ' ')continue;
            else if(*ite == '-')mul = -1;
            else if(*ite == '+')mul = 1;
            else if(*ite < '0' || *ite > '9' )return mul*ans;
            else{
                if(mul*ans < INT_MIN)return INT_MIN;
                else if(mul*ans > INT_MAX) return INT_MAX;
                ans = 10*ans +*ite - '0';
            }
        }
        
        return mul*ans;
    }
};

该代码在914 / 1079 个通过测试用例"+-2"该输出0，结果输出-2，错了，看样子两个‘+-’连在一起是非法的而不是‘-’，我们改代码：

class Solution {
public:
    int myAtoi(string str) {
        string::iterator ite;
        int mul = 1;
        long ans = 0;
        for(ite = str.begin(); ite != str.end() ; ite++)
        {
            if(*ite == ' ')continue;
            else if(*ite == '-')
            {
                if(ite !=str.begin() && (*(ite-1) == '-' || *(ite-1) == '+') )return 0;
                mul = -1;
            }
            else if(*ite == '+')
            {
                if(ite !=str.begin() && (*(ite-1) == '-' || *(ite-1) == '+') )return 0;
                mul = 1;
            }
            else if(*ite < '0' || *ite > '9' )return mul*ans;
            else{
                if(mul*ans < INT_MIN)return INT_MIN;
                else if(mul*ans > INT_MAX) return INT_MAX;
                ans = 10*ans +*ite - '0';
            }
        }
        
        return mul*ans;
    }
};

但是该代码测到917 / 1079 个通过测试用例" +0 123"时失败了，又少考虑情况了，看样子字符串中间的空格是非法的,看样只能先处理字符‘’，‘-’，‘+’,然后再进行判断了，最终代码如下：

代码

class Solution {
public:
    int myAtoi(string str) {
        int n = str.size();
        int i = 0, mul = 1;
        long ans = 0;
        while(str[i] == ' ')i++;//处理空格
        if(str[i] == '-'){
            mul = -1;++i;
        }
        else if(str[i] == '+')++i;
        for(;i < n; i++)
        {
            if(str[i] < '0' || str[i] > '9')return mul*ans;
            else{
                ans = 10*ans + str[i] - '0';
                if(mul*ans < INT_MIN)return INT_MIN;
                else if(mul*ans > INT_MAX) return INT_MAX;                
            }
        }
        return mul*ans;
    }
};