LeetCode(8)—— 字串轉整數 (atoi)
阿新 • • 發佈:2018-12-16
題目內容
描述
實現 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)
函式獲取字元ch
的ASCII碼