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

LeetCode(8)—— 字串轉整數 (atoi)

題目內容

描述

實現 atoi,將字串轉為整數。

在找到第一個非空字元之前,需要移除掉字串中的空格字元。如果第一個非空字元是正號或負號,選取該符號,並將其與後面儘可能多的連續的數字組合起來,這部分字元即為整數的值。如果第一個非空字元是數字,則直接將其與之後連續的數字字元組合起來,形成整數。

字串可以在形成整數的字元後面包括多餘的字元,這些字元可以被忽略,它們對於函式沒有影響。

當字串中的第一個非空字元序列不是個有效的整數;或字串為空;或字串僅包含空白字元時,則不進行轉換。

若函式不能執行有效的轉換,返回 0。

說明

假設我們的環境只能儲存 32 位有符號整數,其數值範圍是 [−2^31, 2^31 − 1]。如果數值超過可表示的範圍,則返回 INT_MAX (2^31 − 1)

INT_MIN (−2^31)

示例:

  • 示例1
輸入: "   -42"
輸出: -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) 。

解法

首先,肯定是將字串的頭尾空格去掉。strip()函式真好用。

然後,我的第一反應是排除非法情況,後來在布林條件配合and和or判斷時有點混亂了 。。。這時不妨換個思路,把一些合法的情況找出來就OK,無非就三種

  • 正號開頭
  • 負號開頭
  • 數字開頭

最後,就是比較常規的“乘10加當前數值”操作,別忘了最後有個確定符號的操作和溢位的判斷。

總體還是easy,不過有一些細節值得記錄

class Solution:
    def myAtoi(self, str):
        """
        :type str: str
        :rtype: int
        """
        # 去除首尾空格
        str = str.strip()
        if str == "":
            return 0
        # 判斷輸入字串是否合法(即是否需要轉換 )
        legal = (ord(str[0]) >= 48 and ord(str[0]) <= 57) or str[0] == '-' or str[0] == '+'
        if not legal:
            return 0
        # 沒有符號,預設為正數
        sign = 1
        if str[0] == '-':
            sign = -1
            str = str[1:]
        elif str[0] == '+':
            str = str[1:]

        # 不斷除以10
        result = 0
        for i in range(len(str)):
            if ord(str[i]) >= 48 and ord(str[i]) <= 57:
                result = 10 * result + (ord(str[i]) - 48)
                continue
            break
        # 溢位判斷
        if result * sign <= -2 ** 31:
            return -2 ** 31
        elif result * sign >= 2 ** 31 - 1:
            return 2 ** 31 - 1
        else:
            return result * sign


s = Solution()
print(s.myAtoi(""))

要注意的地方

  • 去除首尾空格(也可以去除別的字元,給定引數即可)——strip()函式

  • python中布林型變數取反,用not關鍵字。(在別的語言裡面我們會用 !Flag這樣的操作)

  • python中沒有字元相加減,但可以用ord(ch)函式獲取字元chASCII碼