leetcode 8.字符串转换整数(golang)

83 阅读3分钟

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

请你来实现一个 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" 在范围 [-231, 231 - 1] 内,最终结果为 42

示例 2:

输入: s = "   -42"
输出: -42
解释:
第 1 步:" -42"(读入前导空格,但忽视掉)
            ^
第 2 步:"   -42"(读入 '-' 字符,所以结果应该是负数)
             ^
第 3 步:"   -42"(读入 "42")
               ^
解析得到整数 -42 。
由于 "-42" 在范围 [-231, 231 - 1] 内,最终结果为 -42

示例 3:

输入: s = "4193 with words"
输出: 4193
解释:
第 1 步:"4193 with words"(当前没有读入字符,因为没有前导空格)
         ^
第 2 步:"4193 with words"(当前没有读入字符,因为这里不存在 '-' 或者 '+')
         ^
第 3 步:"4193 with words"(读入 "4193";由于下一个字符不是一个数字,所以读入停止)
             ^
解析得到整数 4193 。
由于 "4193" 在范围 [-231, 231 - 1] 内,最终结果为 4193

提示:

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

解答:
这道题没有什么算法逻辑,主要是处理异常情况比较复杂。
首先,要处理前置空格以及+-符号,使用一个res记录结果,将整数部分记录下来之后break出循环。
注意:要处理溢出的情况。

func myAtoi(s string) int {
	res := 0
	flag := 1
	temp := int(math.Pow(2, 31))
	for i, v := range s {
		if res == 0 && (v == '-' || v == '+' || v == ' ') {
			if i > 0 && (s[i-1] == '-' || s[i-1] == '+') {
				break
			}
			if v == '-' {
				flag = -1
			}
		} else if v >= '0' && v <= '9' {
			str := string(v)
			num, _ := strconv.Atoi(str)
			res = res*10 + num
			if i+1 < len(s) && (s[i+1] < '0' || s[i+1] > '9') {
				break
			}
		} else if res == 0 && v != ' ' {
			break
		}
		if res*flag >= (temp - 1) {
			return temp - 1
		}
		if res*flag < -temp {
			return -temp
		}
	}
	if res*flag >= (temp - 1) {
		return temp - 1
	}
	if res*flag < -temp {
		return -temp
	}
	return flag * res
}