1. FlyPython首页
  2. 数据结构与算法
  3. leetcode题解

「LeetCode每日一题」—— 8. 字符串转换整数 (atoi)

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

链接:https://leetcode-cn.com/problems/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) 。

思路

根据题意有三条转化的规则:

  1. 如果第一个非空字符为正或者负号时,则将该符号与之后面尽可能多的连续数字字符组合起来,形成一个有符号整数。
  2. 假如第一个非空字符是数字,则直接将其与之后连续的数字字符组合起来,形成一个整数。
  3. 该字符串在有效的整数部分之后也可能会存在多余的字符,那么这些字符可以被忽略,它们对函数不应该造成影响。

简略来讲就是:

  1. 空格处理
  2. 正负号处理
  3. 数字处理
    还要注意越界处理,代码见暴力解决方案。

在题解中发现了使用一行正则来完成这道题的方案,可参考,见正则表达式方案。

方案代码

暴力解决方案:

class Solution:
    def myAtoi(self, str_: str) -> int:
        flag, ans = None, []
        for c in str_:
            #对空格的处理
            if c == ' ' and not ans and not flag:
                continue
            #对正负号的处理
            elif c in '+-' and not flag and not ans:
                flag = (1 if c=='+' else -1)
            #对数字的处理
            elif c in '0123456789':
                ans.append(c)
            else:
                break
        if not ans:
            return 0

        ans = (flag if flag else 1)*int(''.join(ans))
        #越界处理
        return ans if -2**31<=ans<=2**31-1 else (2**31-1 if ans>0 else -2**31)

正则表达式方案:

class Solution:
    def myAtoi(self, s: str) -> int:
        return max(min(int(*re.findall('^[\+\-]?\d+', s.lstrip())), 2**31 - 1), -2**31)

说明:

^:匹配字符串开头

[+-]:代表一个+字符或-字符
?:前面一个字符可有可无
\d:一个数字
+:前面一个字符的一个或多个
\D:一个非数字字符
*:前面一个字符的0个或多个

Python小技巧

把头像变灰,需要这几行代码

>>> import cv2
>>> img = cv2.imread("input.jpg",cv2.IMREAD_COLOR)
>>> gray  = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
>>> cv2.imwrite("output.jpg",gray)

原创文章,作者:flypython,如若转载,请注明出处:http://flypython.com/algorithm/leetcode/305.html