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

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

如果思路不清晰，将面临许多 corner case 的题目。

题目描述

请你来实现一个 myAtoi(string s) 函数，使其能将字符串转换成一个 32 位有符号整数（类似 C/C++ 中的 atoi 函数）。

函数 myAtoi(string s) 的算法如下：

• 读入字符串并丢弃无用的前导空格
• 检查下一个字符（假设还未到字符末尾）为正还是负号，读取该字符（如果有）。 确定最终结果是负数还是正数。 如果两者都不存在，则假定结果为正。
• 读入下一个字符，直到到达下一个非数字字符或到达输入的结尾。字符串的其余部分将被忽略。
• 将前面步骤读入的这些数字转换为整数（即，"123" -> 123， "0032" -> 32）。如果没有读入数字，则整数为 0 。必要时更改符号（从步骤 2 开始）。
• 如果整数数超过 32 位有符号整数范围 [−231,  231 − 1] ，需要截断这个整数，使其保持在这个范围内。具体来说，小于 −231 的整数应该被固定为 −231 ，大于 231 − 1 的整数应该被固定为 231 − 1 。
• 返回整数作为最终结果。

注意：

• 本题中的空白字符只包括空格字符 ' ' 。
• 除前导空格或数字后的其余字符串外，请勿忽略 任何其他字符。

提示：

• 0 <= s.length <= 200
• s 由英文字母（大写和小写）、数字（0-9）、' '、'+'、'-' 和 '.' 组成

解题思路

从头看需要考虑的 corner case 很多，比如 0-123 0 123 等都非常容易出错。

梳理一下我们的思路，发现题目中输入字符的合法排列，只有 空格、正负号、前导0、整数 这一种合法的排列，遇到非法的字符立即返回0即可。按照这种思路来组织逻辑，我们的代码将会非常清晰。

参考代码

class Solution {
public:
    int myAtoi(string s) {
        size_t n = s.size();
        size_t i = 0;
        while (i < n) {
            if (s[i] == ' ') i ++;
            else break;
        }
        int flag = 1;
        if (i < n && (s[i] == '-' || s[i] == '+')) {
            flag = (s[i] == '-') ? -1 : 1;
            i ++;
        }
        while (i < n) {
            if (s[i] == '0') i ++;
            else break;
        }
        int64_t res = 0;
        while (i < n) {
            if (!isdigit(s[i])) break;
            res = res * 10 + s[i] - '0';
            if (res*flag > INT32_MAX || res*flag < INT32_MIN) break;
            i++;
        }
        res *= flag;
        if (res < INT32_MIN) return INT32_MIN;
        if (res > INT32_MAX) return INT32_MAX;
        return res;
    }
};