文章目錄
-
- 題目
- 解題思路
- 正規表達式解法
題目
題目來源
請你來實作一個 atoi 函數，使其能将字元串轉換成整數。
首先，該函數會根據需要丢棄無用的開頭空格字元，直到尋找到第一個非空格的字元為止。接下來的轉化規則如下：
- 如果第一個非空字元為正或者負号時，則将該符号與之後面盡可能多的連續數字字元組合起來，形成一個有符号整數。
- 假如第一個非空字元是數字，則直接将其與之後連續的數字字元組合起來，形成一個整數。
- 該字元串在有效的整數部分之後也可能會存在多餘的字元，那麼這些字元可以被忽略，它們對函數不應該造成影響。
注意：假如該字元串中的第一個非空格字元不是一個有效整數字元、字元串為空或字元串僅包含空白字元時，則你的函數不需要進行轉換，即無法進行有效轉換。
在任何情況下，若函數不能進行有效的轉換時，請傳回 0 。
提示：
- 本題中的空白字元隻包括空格字元 ’ ’ 。
- 假設我們的環境隻能存儲 32 位大小的有符号整數，那麼其數值範圍為 [−231, 231 − 1]。如果數值超過這個範圍，請傳回 INT_MAX (231 − 1) 或 INT_MIN (−231) 。
示例：
輸入: “42”
輸出: 42
輸入: " -42"
輸出: -42
解釋: 第一個非空白字元為 ‘-’, 它是一個負号。我們盡可能将負号與後面所有連續出現的數字組合起來，最後得到 -42 。
輸入: “4193 with words”
輸出: 4193
解釋: 轉換截止于數字 ‘3’ ，因為它的下一個字元不為數字。
輸入: “words and 987”
輸出: 0
解釋: 第一個非空字元是 ‘w’, 但它不是數字或正、負号。是以無法執行有效的轉換。
輸入: “-91283472332”
輸出: -2147483648
解釋: 數字 “-91283472332” 超過 32 位有符号整數範圍。是以傳回 INT_MIN (−231) 。
解題思路
根據題目要求，能輸出整數結果的可能字元串有兩種：1.字元串中第一個數字前隻有空格或什麼都沒有；2.第一個數字前為“+”或“-”，符号前隻有空格或什麼都沒有。具體情況如下圖：
是以，解本題步驟如下：
- 從左開始删除空格字元，直到出現第一個非空字元；
- 判斷第一個非空字元是否為數字或正負号，若為False，則直接傳回0，若為True，則繼續下一步；另外當字元串為空時，也傳回0；
- 逐個周遊，并将第一個非空字元後所有連續出現的數字組合成新的字元串；
- 使用int()函數，将新的字元串轉換成整數；
- 若整數在數值範圍[−231, 231 − 1]，則傳回整數，否則傳回−231或 231 − 1。
代碼實作：
class Solution:
def myAtoi(self, str: str) -> int:
str = str.lstrip() # 從左開始查找删除空格字元
"""如果字元串為空，或者第一個非空字元不是數字或["+","-"]，則傳回0"""
if len(str) == 0 or (str[0].isdigit() == False and str[0] not in ["+","-"]):
return 0
res,i = str[0],1
while i < len(str) and str[i].isdigit():
res += str[i]
i += 1
try:
res = int(res)
return min(max(res,-2**31),2**31-1) # 判斷數值是否在區間[−2**31,2**31 − 1]
except:
return 0
正規表達式解法
通常使用if語句會寫很多個，且容易有遺漏的特殊情況，而使用正規表達式方法，可避免寫一堆if語句。
代碼實作：
class Solution:
def myAtoi(self, str: str) -> int:
import re
re_str = r"\s*[-|+]?\d+"
result = re.match(re_str, str)
try:
result = int(result[0].strip(" "))
except:
return 0
else:
return max(min(result, 2**31 - 1), (-2)**31)
最短一行代碼實作：
class Solution:
def myAtoi(self, s: str) -> int:
return max(min(int(*re.findall('^[\+\-]?\d+', s.lstrip())), 2**31 - 1), -2**31)
Python 1 行 為什麼可以用正規表達式？