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

难度:中等

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/string-to-integer-atoi/

1. 问题描述

请你来实现一个 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. 返回整数作为最终结果。

注意:

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

2. 分析与解答

思路:字符串解析。分情况:

  • 去除字符串开始的空格
  • 判断正负号
  • 逐个判断是否为数字并将其转换成数字,注意判断是否越界

注意:中间遇到空格也是要退出的情形

代码:

class Solution {
public:
    int myAtoi(string s) {
        int n = s.length();
        int res = 0;
        int index = 0;
        // 去除前置空格
        while (index < n) {
            if (s[index] == ' ') index++;
            else break;
        }
        // 从第一个非空格开始,判断是否是负数
        int p = 1;
        if (s[index] == '-') {
            p = -1;
            index ++;
        } else if (s[index] == '+') {
            index ++;
        }
        for (int i = index; i < n; i++) {
            if (isdigit(s[i])) {
                int add = s[i]-'0';
                if (res > (INT_MAX - add) / 10) {
                    return p == 1 ? INT_MAX : INT_MIN;
                } else {
                    res = res * 10 + add;
                }
            } else break;
        }
        return res * p;
    }
};