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

题目描述

请你来实现一个 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) 。

解法

func myAtoi(str string) int {
	var (
		f      int
		numStr string
		num    int
		k      int
	)
	for {
		if len(str) < 1 {
			return 0
		}
		ch := str[0]
		if ch == 32 {
			str = str[1:]
		} else {
			break
		}
	}
	ch := str[0]
	if ch == 45 {
		str = str[1:]
		f = -1
	} else if ch == 43 {
		str = str[1:]
		f = 1
	} else if ch >= 48 && ch <= 57 {
		f = 1
	} else {
		f = 0
	}
	if f == 0 {
		return 0
	} else {
		for index := 0; index < len(str); index++ {
			if str[index] >= 48 && str[index] <= 57 {
				numStr += string(str[index])
			} else {
				break
			}
		}
	}
	newNumLength := len(numStr)
	for index := 0; index < newNumLength; index++ {
		if k != 0 {
			num *= 10
		}
		num += int(numStr[index]) - 48
		k++
		if num*f >= 2147483648 {
			return 2147483647
		}
		if num*f < -2147483648 {
			return -2147483648
		}
	}
	num = num * f
	return num
}

本地测试

package main

import (
	"fmt"
)

func main() {
	s := "42"
	fmt.Println(myAtoi(s), 42)
	s = " "
	fmt.Println(myAtoi(s), 0)
	s = "  "
	fmt.Println(myAtoi(s), 0)
	s = ""
	fmt.Println(myAtoi(s), 0)
	s = "-42"
	fmt.Println(myAtoi(s), -42)
	s = " -42"
	fmt.Println(myAtoi(s), -42)
	s = "+42"
	fmt.Println(myAtoi(s), 42)
	s = "0"
	fmt.Println(myAtoi(s), 0)
	s = "9"
	fmt.Println(myAtoi(s), 9)
	s = "4193 with words"
	fmt.Println(myAtoi(s), 4193)
	s = "words and 987"
	fmt.Println(myAtoi(s), 0)
	s = "-91283472332"
	fmt.Println(myAtoi(s), -2147483648)
	s = "2147483648"
	fmt.Println(myAtoi(s), 2147483647)
	s = "9223372036854775808"
	fmt.Println(myAtoi(s), 2147483647)
}

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/string-to-integer-atoi