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

148 阅读2分钟

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

函数 myAtoi(string s) 的算法如下:

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

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/st… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法:

import "math"
import "strconv"
func myAtoi(s string) int {
    ans := int32(0)
    skipedSpace := false
    skipedFlag := false
    flag := 1
    for i := range s {
        if !skipedSpace && s[i] == byte(' ') {
            continue
        } 
        skipedSpace = true
        if !skipedFlag {
            skipedFlag = true
            if s[i] == byte('-') {
                flag = -1
                continue
            } else if s[i] == byte('+') {
                continue
            }
            
        }

        // 计算数字,如果不能用strconv的话就用map做转换好了
        n, err := strconv.Atoi(string(s[i]))
        if err != nil {
            break
        }
        last := ans
        ans = ans * 10 + int32(n)
        if ans / 10 != last {
            if flag == 1 {
                return int(math.MaxInt32)
            } else {
                return int(math.MinInt32)
            }
        }
    }
    return int(ans) * flag
}