1. 程式人生 > >【LeetCode】Python 8.字串轉整數 (atoi)

【LeetCode】Python 8.字串轉整數 (atoi)

字串轉整數 (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) 。

程式碼:

class Solution:
    def myAtoi(self, str):
        """
        :type str: str
        :rtype: int
        """
        # 去掉非空字元前的空字元
        for i in str:
            if i == " ":
                  str = str.replace(i, ""
, 1) else: break # 目標字串 ret_str = "" # 字串為空,或不是有效整數,返回0 if str == "": return 0 elif str[0] == "+" or str[0] == "-" or str[0].isdecimal(): ret_str += str[0] else: return 0 # 將盡可能多的連續數字組合起來 for i in str[1:]: if i.isdecimal(): ret_str += i # 數值範圍是 [−231, 231 − 1] if int(ret_str) > 2**31-1: return 2147483647 elif int(ret_str) < -2**31: return -2147483648 else: break if ret_str == "+" or ret_str == "-": return 0 else: ret_str = int(ret_str) return ret_str