【面試題67 把字符串轉換成整數】
面試題67:把字符串轉換成整數:寫一個函數 StrToInt,實現把字符串轉換成整數這個功能。不能使用 atoi 或者其他類似的庫函數。
首先,該函數會根據需要丟棄無用的開頭空格字符,直到尋找到第一個非空格的字符爲止。當我們尋找到的第一個非空字符爲正或者負號時,則將該符號與之後面儘可能多的連續數字組合起來,作爲該整數的正負號;假如第一個非空字符是數字,則直接將其與之後連續的數字字符組合起來,形成整數。該字符串除了有效的整數部分之後也可能會存在多餘的字符,這些字符可以被忽略,它們對於函數不應該造成影響。
注意:假如該字符串中的第一個非空格字符不是一個有效整數字符、字符串爲空或字符串僅包含空白字符時,則你的函數不需要進行轉換。在任何情況下,若函數不能進行有效的轉換時,請返回 0。
假設我們的環境只能存儲 32 位大小的有符號整數,那麼其數值範圍爲 [−231, 231 − 1]。如果數值超過這個範圍,請返回 INT_MAX (231 − 1) 或 INT_MIN (−231) 。
思路:去掉首尾空格後,先判斷第一個字符,是符號位則記錄一下,用於後面轉負數,然後從第二個字符開始循環;是數字就開始循環,直到非數字 break 掉;其餘情況都是無效字符串直接 return 0。用 res 數組記錄數字,循環結束後判斷 res 是否有數字,有就加上符號位轉數值,否則 return 0。最後注意用 int_max 和 int_min 判斷是否越界。
特例判斷:首先用 strip() 去掉首尾空格
- 若字符串長度爲 0,無效,直接返回 0
- 若第一個字符非符號、非數字,無效,直接返回 0
時間複雜度:,最壞情況下,遍歷字符串一遍
空間複雜度: ,需要 res 來存儲數字字符,最壞情況下,所有字符全爲數字
class Solution:
def strToInt(self, str: str) -> int:
str = str.strip()
if not str: return 0
n = len(str)
str = list(str)
number = ['0','1','2','3','4','5','6','7','8','9']
sign = ['-', '+']
res = []
if str[0] not in number and str[0] not in sign: return 0
int_max = pow(2, 31) - 1
int_min = -pow(2, 31)
flag = 0
if str[0] in sign:
flag = 1 if str[0] == '-' else 0
for i in range(1, n):
if str[i] in number: res.append(str[i])
else: break
else:
for i in range(n):
if str[i] in number: res.append(str[i])
else: break
if res:
res = int(''.join(res))
else: return 0
if flag == 1:
res = -res
if res < int_min: return int_min
if res > int_max: return int_max
return res