Leetcode 8.字符串转整数

题目

8.字符串转整数

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

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

注意：

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

示例 1：

输入：s = “42”
输出：42
解释：加粗的字符串为已经读入的字符，插入符号是当前读取的字符。
第 1 步： “42”（当前没有读入字符，因为没有前导空格）
      ^
第 2 步：“42”（当前没有读入字符，因为这里不存在 ‘-’ 或者 ‘+’）
      ^
第 3 步：“42”（读入 “42”）
      ^
解析得到整数 42 。
由于 “42” 在范围 [-2³¹, 2³¹ - 1] 内，最终结果为 42 。

示例 2：

输入：s = " -42"
输出：-42
解释：
第 1 步：" -42"（读入前导空格，但忽视掉）
      ^
第 2 步：" -42"（读入 ‘-’ 字符，所以结果应该是负数）
        ^
第 3 步：" -42"（读入 “42”）
        ^
解析得到整数 -42 。
由于 “-42” 在范围 [-2³¹, 2³¹ - 1] 内，最终结果为 -42 。

示例 3：

输入：s = “4193 with words”
输出：4193
解释：
第 1 步：“4193 with words”（当前没有读入字符，因为没有前导空格）
      ^
第 2 步：“4193 with words”（当前没有读入字符，因为这里不存在 ‘-’ 或者 ‘+’）
      ^
第 3 步：“4193 with words”（读入 “4193”；由于下一个字符不是一个数字，所以读入停止）
       ^
解析得到整数 4193 。
由于 “4193” 在范围 [-2³¹, 2³¹ - 1] 内，最终结果为 4193 。

提示：

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

思路

注意子串和子序列的区别：

• 子串一定是连续的一段；
• 子序列不一定是连续的一段，但要求下标是递增的；

算法：
模拟算法 O(n)

1、找到第一个非空的字符位置k，若k == n表示全都是空字符，直接返回0
2、op 记录截取后的整数的正负性，1表示正号，-1表示负号
3、使用 int 类型的 res 来存储截取后的整数（一定是正整数，因为括号还没包括），
在某一刻若 op 是正数，(res * 10 + x) * op > INT_MAX，即 res * op > (INT_MAX - x ) / 10 ，则返回INT_MAX，表示已经超出（其中x是正数）
在某一刻若op是负数，(res * 10 + x) * op < INT_MIN，即 res * op > (INT_MIN - x * (-1)) / 10，则返回INF_MIN，表示已经超出（其中x是负数）
4、截取到 res 后，附上 op 的符号，若 res > INT_MAX 或者 res < INT_MIN 则分别返回 INF_MAX 或者INT_MIN
5、最后返回 res

时间复杂度： 每个字符最多遍历一次，所以总时间复杂度是 O(n) .

代码

C++代码：

class Solution {
public:
    int myAtoi(string str) {
        int res = 0;
        int k = 0;
        while (k < str.size() && str[k] == ' ') k ++ ;
        if (k == str.size()) return 0;

        int minus = 1;
        if (str[k] == '-') minus = -1, k ++ ;
        if (str[k] == '+')
            if (minus == -1) return 0;
            else k ++ ;

        while (k < str.size() && str[k] >= '0' && str[k] <= '9') {
            int x = str[k] - '0';
            if (minus > 0 && res > (INT_MAX - x) / 10) return INT_MAX;
            if (minus < 0 && -res < (INT_MIN + x) / 10) return INT_MIN;
            if (-res * 10 - x == INT_MIN) return INT_MIN;
            res = res * 10 + x;
            k ++ ;
        }

        res *= minus;
        return res;
    }
};

python3代码：

class Solution:
    def myAtoi(self, s: str) -> int:
        idx, sign = 0, 1
        while idx < len(s) and s[idx] == " ": idx += 1

        if idx < len(s) and s[idx] == "+": idx += 1
        elif idx < len(s) and s[idx] == "-": idx += 1; sign = -1

        num = 0
        while idx < len(s) and s[idx] >= '0' and s[idx] <= '9':
            num = num * 10 + int(s[idx])
            idx += 1

        num *= sign
        maxv = (1 << 31)
        if num > 0 and num > maxv - 1: num = maxv - 1
        if num < 0 and num < -maxv: num = - maxv

        return num