leetcode题解系列-007 字符串转换整数 (atoi)

老规矩，先上代码，后续有解题思路。

//
// Created by tannzh on 2020/6/15.
//

/*
 * 字符串转换整数 (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) 。

 */

#include <string>
#include <climits>
#include <iostream>

using namespace std;

class Solution {
public:
    int myAtoi(string str) {
        int n = str.size();
        int i;
        int flag = 0;
        long ans = 0;
        for(i = 0 ; i < n ; i ++){
            if(flag == 0){
                if(str[i] == '+') flag = 1;
                else if(str[i] == '-') flag = -1;
                else if(str[i] == ' ')   continue;
                else if(str[i] >= '0' && str[i] <= '9'){
                    ans = str[i] - '0';
                    flag = 1;
                }
                else return 0;
            }
            else{
                if(str[i] >= '0' && str[i] <= '9'){
                    ans = ans * 10 + (str[i] - '0');
                    if(ans * flag < INT_MIN) return INT_MIN;
                    if(ans * flag > INT_MAX) return INT_MAX;
                }
                else if(ans != 0)    break;
                else return 0;
            }
        }
        return ans * flag;
    }
};

int main(int argc, char **argv)
{
    Solution s;
    std::string str1 = "42";
    std::string str2 = "-42";
    std::string str3 = "4193 with words";
    std::string str4 = "words and 987";
    std::string str5 = "-91283472332";

    std::cout << "input: " << str1 << ", output: " << s.myAtoi(str1) << std::endl;
    std::cout << "input: " << str2 << ", output: " << s.myAtoi(str2) << std::endl;
    std::cout << "input: " << str3 << ", output: " << s.myAtoi(str3) << std::endl;
    std::cout << "input: " << str4 << ", output: " << s.myAtoi(str4) << std::endl;
    std::cout << "input: " << str5 << ", output: " << s.myAtoi(str5) << std::endl;

    return 0;
}

解题思路

 

和上一题解题方法相似，使用long类型变量。需要考虑的情况较多，类似于自动机，存在多个状态。另外需要处理一些边界情况，理解题意也很重要。