1. 程式人生 > >leetcode 字串轉整數 (atoi)

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

分析:
題目也比較簡單,按照題目的規則進行轉換。
如果第一個非空白符不是數字,或者字串為空,或者全是字母,則結果為0,否則進行轉換;
注意可能存在正負號,如果數字超過int型最大最小值範圍,自返回對應的最大最小值;

python程式碼實現:

#!/usr/bin/env python3
# -*- coding : utf-8 -*-

import re
class Solution:
    def myAtoi(self, str):
        """
        :type str: str
        :rtype: int
        """
        str = str.strip()
        m = re.match(r'^([-+]?[0-9]+).*', str)
        if not m:
            return 0
        num_of_str = m.group(1)
        num = int(num_of_str)
        if num < 0 and num < (-2 ** 31):
            return -2**31
        if num > 0 and num > 2**31 - 1:
            return 2**31 - 1
        return num

if __name__ == '__main__':
    s = Solution()
    print(s.myAtoi("42"))
    print(s.myAtoi(""))
    print(s.myAtoi("     -42"))
    print(s.myAtoi("4223 with word"))
    print(s.myAtoi("word with 42"))
    print(s.myAtoi("-912834723320000000"))
    print(s.myAtoi("91283472332"))

C++程式碼實現:

#include<iostream>
#include<string>
#include<limits.h>
using namespace std;
class Solution {
public:
    int myAtoi(string str) {
        int len = str.size();
        if(str.size() == 0){
            return 0;
        }
        int i = 0;
        while(i<len && str[i] == ' ')i++;
        if(i>=len)return 0;
        int sign = 1;
        if(str[i] == '-'){
            sign = -1;
            i++;
        } else if(str[i]=='+'){
            i++;
        }
        if(str[i]<'0' || str[i]>'9'){
            return 0;
        }
        int num = 0;
        while(i<len && str[i]>='0' && str[i]<='9'){
            int add=str[i]-'0';
            if(sign == 1){
                if((INT_MAX-add)/10.0<num)return INT_MAX;
            } else {
                if((-INT_MIN+add)/10.0>-num)return INT_MIN;
            }
            num = num*10 + add;
            i++;
        }
        return sign*num;
    }
};
int main(){
    Solution s;
    cout<<s.myAtoi("42")<<endl;
    cout<<s.myAtoi("   -42")<<endl;
    cout<<s.myAtoi("42 with world")<<endl;
    cout<<s.myAtoi("with world 42")<<endl;
    cout<<s.myAtoi("-91283472332")<<endl;

    return 0;
}