问题描述
请你来实现一个 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) 。
思路
这一题就是细节问题,边界问题特别多,注意一下就好了;
这一题可以先写个简单版,再根据题意逐一处理;
AC代码
class Solution {
public:
int myAtoi(string s) {
int k = 0;
if (s[k] >= 'a' && s[k] <= 'z' && s[k] >= 'A' && s[k] <= 'Z') return 0;
while (k < s.size() && s[k] == ' ') k ++;
if (k == s.size()) return 0;
long long res = 0;
int t = 1;
if (k < s.size()) {
if (s[k] == '-') t = -1, k ++;
else if (s[k] == '+') k ++;
}
while (k < s.size() && s[k] >= '0' && s[k] <= '9') {
res = res * 10 + s[k] - '0';
k ++;
if (res > INT_MAX) break; // res必为正数所以判断一下
}
res *= t;
if (res > INT_MAX) return INT_MAX;
if (res < INT_MIN) return INT_MIN;
return res;
}
};
另一种版本不用long long来存储
第21行是因为INT_MIN最小值是-2147483648而INT_MAX是2147483647,所以可能会越界,要特判一下
class Solution {
public:
int myAtoi(string s) {
int k = 0;
if (s[k] >= 'a' && s[k] <= 'z' && s[k] >= 'A' && s[k] <= 'Z') return 0;
while (k < s.size() && s[k] == ' ') k ++;
if (k == s.size()) return 0;
int t = 1;
if (k < s.size()) {
if (s[k] == '-') t = -1, k ++;
else if (s[k] == '+') k ++;
}
int res = 0;
while (k < s.size() && s[k] >= '0' && s[k] <= '9') {
int x = s[k] - '0';
if (t > 0 && res > (INT_MAX - x) / 10) return INT_MAX;
if (t < 0 && -res < (INT_MIN + x) / 10) return INT_MIN;
if (-res * 10 - x == INT_MIN) return INT_MIN;
res = res * 10 + x;
k ++;
}
res *= t;
return res;
}
};