LeetCode之（8）字符串转换整数 (atoi)（String to Integer（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) 。

题目分析

首先需要做的是从输入字符串中提取有效的整数部分，然后将有效的整数部分转换成int型，并根据情况输出。在中间过程使用字符串来存储与处理。

用例：

 In:   123 4
Out:123
 In:   -
Out:0
 In:  +123
Out:123
 In:   00000000000123
Out:123
 In:  12345678901
Out:2147483647
 In:-2147483648
Out:-2147483648
 In:30141
Out:3

代码实现

C++

/*
 * @lc app=leetcode.cn id=8 lang=cpp
 *
 * [8] 字符串转换整数 (atoi)
 */

// @lc code=start
class Solution {
public:
    int myAtoi(string str) {
        //预处理提取数字部分
        string input = prepare(str);
        if(input.size() == 0 || input == "-" ){
            return 0;
        }
        //判断正负
        int sign = 1;
        if(input[0] == '-'){
            sign = -1;
            input = input.substr(1,input.size() - 1);
        }
        else if(input[0] == '+'){
            sign = 1;
            input = input.substr(1,input.size() - 1);
        }

        //全是0
        int i;
        for(i = 0; i < input.size(); i++){
            if(input[i] == '0' ){
                continue;
            }
            else{
                break;
            }
        }
        if(i == input.size()){
            return 0;
        }
        input = input.substr(i,input.size()-i);
        string max_str = "2147483647";
        string min_str = "2147483648";
        int max = 2147483647;
        int min = -2147483648;
        //长度超出范围
        if(input.size() > 10){
            if(sign == 1){
                return max;
            }
            else{
                return min;
            }
        }
        //长度为10，单独处理
        if(input.size() == 10){
            if(sign == 1){
                if(input > max_str){
                    return max;
                }
            }
            else if(sign == -1 ){
                if(input > min_str){
                    return min;
                }
            }
        }
        //长度小于10
        int ans = 0;
        for(int i = 0; i < input.size(); i++){
            ans = ans*10 + (input[i] - '0')*sign;
        }
        return ans;
    }
    string prepare(string str){
        if(str.size() == 0){
            return str;
        }
        int start, end;
        // the position of the number' start 
        for(int i = 0;i < str.size(); i++){
            if(str[i] == ' '){
                continue;
            }
            else if(str[i] == '-' || str[i] == '+' || (str[i] <= '9' && str[i] >= '0') ){
                start = i;
                break;
            }
            else{
                return "";
            }
        }

        // the position of the number' end
        end = start;
        for(int i = start + 1; i < str.size(); i++){
            if(str[i] <= '9' && str[i] >= '0'){
                end = i;
            }
            else{
                break;
            }
        }
        return str.substr(start, end - start + 1);
    }
};
// @lc code=end

Java

/*
 * @lc app=leetcode.cn id=8 lang=java
 *
 * [8] 字符串转换整数 (atoi)
 */

// @lc code=start
class Solution {
    public int myAtoi(String str) {
        String input = prepare(str);
        if(input.length() == 0 || input.compareTo("-") == 0 || input.compareTo("+") == 0){
            return 0;
        }

        int sign = 1;
        if(input.charAt(0) == '-'){
            sign = -1;
            input = input.substring(1);
        }
        else if(input.charAt(0) == '+'){
            sign = 1;
            input = input.substring(1);
        }

        int i;
        for(i = 0; i < input.length(); i++){
            if(input.charAt(i) == '0' ){
                continue;
            }
            else{
                break;
            }
        }
        if(i == input.length()){
            return 0;
        }

        input = input.substring(i);
        String max_str = "2147483647";
        String min_str = "2147483648";
        int max = 2147483647;
        int min = -2147483648;

        if(input.length() > 10){
            if(sign == 1){
                return max;
            }
            else{
                return min;
            }
        }

        if(input.length() == 10){
            if(sign == 1){
                if(input.compareTo(max_str) >= 0){
                    return max;
                }
            }
            else if(sign == -1 ){
                if(input.compareTo(min_str) >= 0){
                    return min;
                }
            }
        }

        return Integer.parseInt(input)*sign;
    }
    private String prepare(String str){
        if(str.length() == 0){
            return str;
        }
        int start = 0, end = 0;
        int i = 0;
        for(i = 0;i < str.length(); i++){
            if(str.charAt(i) == ' '){
                continue;
            }
            else if(str.charAt(i) == '-' || str.charAt(i) == '+' || (str.charAt(i) <= '9' && str.charAt(i) >= '0') ){
                start = i;
                break;
            }
            else{
                return "";
            }
        }
        if(i == str.length()){
            return "";
        }

        end = start;
        for(i = start + 1; i < str.length(); i++){
            if(str.charAt(i) <= '9' && str.charAt(i) >= '0'){
                end = i;
            }
            else{
                break;
            }
        }
        return str.substring(start, end + 1);
    }
}
// @lc code=end

Python

#
# @lc app=leetcode.cn id=7 lang=python3
#
# [7] 整数反转
#

# @lc code=start
class Solution:
    def reverse(self, x: int) -> int:
        sign = 1
        if(x < 0):
            sign = -1
            x = -x
        x_reverse = int(str(x)[::-1])*sign
        if(x_reverse >= -2147483648 and x_reverse <= 2147483647):
            return x_reverse
        else:
            return 0
# @lc code=end

#
# @lc app=leetcode.cn id=8 lang=python3
#
# [8] 字符串转换整数 (atoi)
#

# @lc code=start
class Solution:
    def myAtoi(self, str: str) -> int:
        input = self.prepare(str)
        if(len(input) == 0 or input == "-" or input == '+'):
            return 0

        output = int(input)
        if(output >= -2147483648 and output <= 2147483647):
            return output
        elif(output < -2147483648):
            return -2147483648
        else:
            return 2147483647

    def prepare(self, str:str) -> str:
        if(len(str) == 0):
            return ""
        start = 0
        end = 0
        flag = 0
        for start in range(len(str)):
            if(str[start] == ' '):
                continue
            elif(str[start] == '-' or str[start] == '+' or (str[start] >= '0' and str[start] <= '9')):
                flag = 1
                break
            else:
                return ""
        if(flag == 0):
            return ""
        end = start
        for i in range(start+1,len(str)):
            if(str[i] >= '0' and str[i] <= '9'):
                end = i
            else:
                break
        return str[start:end+1]

# @lc code=end