请你来实现一个 myAtoi(string s) 函数,使其能将字符串转换成一个 32 位有符号整数(类似 C/C++ 中的 atoi 函数)。
函数 myAtoi(string s) 的算法如下:
- 读入字符串并丢弃无用的前导空格
- 检查下一个字符(假设还未到字符末尾)为正还是负号,读取该字符(如果有)。 确定最终结果是负数还是正数。 如果两者都不存在,则假定结果为正。
- 读入下一个字符,直到到达下一个非数字字符或到达输入的结尾。字符串的其余部分将被忽略。
- 将前面步骤读入的这些数字转换为整数(即,"123" -> 123, "0032" -> 32)。如果没有读入数字,则整数为 0 。必要时更改符号(从步骤 2 开始)。
- 如果整数数超过 32 位有符号整数范围 [
,
] ,需要截断这个整数,使其保持在这个范围内。具体来说,小于
的整数应该被固定为
,大于
的整数应该被固定为
。
- 返回整数作为最终结果。
注意:
- 本题中的空白字符只包括空格字符
' '
。 - 除前导空格或数字后的其余字符串外,请勿忽略 任何其他字符。
示例 1:
输入:s = "42"
输出:42
解释:加粗的字符串为已经读入的字符,插入符号是当前读取的字符。
第 1 步:"42"(当前没有读入字符,因为没有前导空格)
^
第 2 步:"42"(当前没有读入字符,因为这里不存在 '-' 或者 '+')
^
第 3 步:"42"(读入 "42")
^
解析得到整数 42 。
由于 "42" 在范围 [-2^31, 2^31 - 1] 内,最终结果为 42 。
示例 2:
输入:s = " -42"
输出:-42
解释:
第 1 步:" -42"(读入前导空格,但忽视掉)
^
第 2 步:" -42"(读入 '-' 字符,所以结果应该是负数)
^
第 3 步:" -42"(读入 "42")
^
解析得到整数 -42 。
由于 "-42" 在范围 [-2^31, 2^31 - 1] 内,最终结果为 -42 。
示例 3:
输入:s = "4193 with words"
输出:4193
解释:
第 1 步:"4193 with words"(当前没有读入字符,因为没有前导空格)
^
第 2 步:"4193 with words"(当前没有读入字符,因为这里不存在 '-' 或者 '+')
^
第 3 步:"4193 with words"(读入 "4193";由于下一个字符不是一个数字,所以读入停止)
^
解析得到整数 4193 。
由于 "4193" 在范围 [-2^31, 2^31 - 1] 内,最终结果为 4193 。
提示:
0 <= s.length <= 200
-
s
由英文字母(大写和小写)、数字(0-9
)、' '
、'+'
、'-'
和'.'
组成
解题思路:动态规划
func myAtoi(s string) int {
if len(s) == 0 {
return 0
}
i, res, flag := 0, 0, 1
// trim
for s[i] == ' ' {
i++
if i >= len(s) {
return 0
}
}
// 判断正负
if s[i] == '-' {
flag= -1
i++
} else if s[i] == '+'{
i++
}
for i < len(s) && s[i] >= '0' && s[i] <= '9' {
res = res*10 + int(s[i]-'0')
i++
if res > math.MaxInt32 {
if flag > 0 {
return math.MaxInt32
} else {
return math.MinInt32
}
}
}
return res * flag
}
但其实一次循环就可以获得最终结果:
func myAtoi(str string) int {
var ans int64 = 0
var sign int64 = 1
start := false
for _, b := range str {
if b <= '9' && b >= '0' {
if !start {start = true}
ans = ans * 10 + int64(b) - int64('0')
if ans * sign > math.MaxInt32 {
return math.MaxInt32
} else if ans * sign < math.MinInt32 {
return math.MinInt32
}
} else if !start && b == '+' {
start = true
} else if !start && b == '-' {
start = true
sign = -1
} else if !start && b == ' '{
continue
} else {
break
}
}
return int(ans * sign)
}
网友评论