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

最新推荐文章于 2022-07-18 16:04:56 发布

原创最新推荐文章于 2022-07-18 16:04:56 发布 · 264 阅读

0 ·

CC 4.0 BY-SA版权

本文详细解析了C++和Python中atoi函数的实现原理，通过具体代码展示了如何从字符串中读取整数值，包括处理前导空白、正负号及溢出情况。

C++代码：

#define INT_MIN -pow(2, 31)
#define INT_MAX pow(2,31)-1
class Solution {
public:
    int myAtoi(string str) {
        // step1: 先把str左边空字符去除 
        int idx = 0;
        for(char ch:str){
            if(ch==' '){
                idx++;
            }else{
                break;
            }
        }
        string ss = str.substr(idx,str.length()-idx);
        // step2: 判断ss第一个字符，根据ss开头来做快速判断
        //简单判断，空字符串返回0； 
        //非空字符串，开头不是'+'，'-'，或数字，直接返回0
        //如果字符串开头是'+', flag = 1; 
        //               '-', flag = -1;
        //               数字， 加入结果数
        if (ss=="") return 0;//空字符串，直接返回0
        int flag = 1;
        double result = 0;
        if (ss[0]=='-'){
            flag = -1;
        }else if('0'<=ss[0] && ss[0]<='9'){
            result += (ss[0]-48);
        }else if(ss[0]=='+'){
            flag = 1;
        }else{
            return 0;
        }

        //step3: 继续将后续连续的数字加入结果数
        ss = ss.substr(1, ss.length()-1);
        for(char ch : ss){
            if ('0'<=ch && ch<='9'){
                result = result*10 + (ch-48);
            }else{
                break; // 退出循环
            }
        }
        //step4: 最后判断一些边界条件，直接输出
        result = result*flag;
        if (result < INT_MIN){
            return INT_MIN;
        }else if(result > INT_MAX){
            return INT_MAX;
        }else{
            return (int)result;
        }
    }
};

Python3代码思路同C++:

INT_MIN = -2**31
INT_MAX = 2**31 - 1
class Solution:
    def myAtoi(self, str: str) -> int:
        ss = str.strip()#先把两边空字符去除
        if ss=="": return 0#空字符串直接返回0
        result = ""#定义结果字符串
        if ss[0]=="+" or ss[0]=="-" or ss[0].isdigit():#第一个字符是这三种情况的话加入结果字符串
            result += ss[0]
        else:
            return 0#如果第一个字符不是上面那三种情况，直接返回0
        for s in ss[1:]:#将后续是数字的加入结果字符串
            if s.isdigit():
                result += s
            else:
                break
        if len(result)==1 and (result[0]=="+" or result[0]=="-"):#经过上面的叠加，最后结果字符串只有一位，并且是"+" 或"-"这种符号位的话，还是没法转换，直接返回0
            return 0
        if int(result) < INT_MIN:#边界条件判断
            return INT_MIN
        elif int(result) > INT_MAX:
            return INT_MAX
        else:
            return int(result)#正常返回