将整数数字字符串转为整数值

题目

给定一个字符串s，如果s符合日常书写的整数形式，并属于32位整数的范围，返回s代表的整数，否则返回0.
例如

• ‘123’ -> 123
• ‘023’ -> 0
• ‘A13’ -> 0
• ‘0’ -> 0
• ‘2147483647’ -> 2147483647
• ‘2147483647’ -> 0 (溢出）
• ‘-123’ -> -123

思路

首先检查s是不是日常书写形式的整数：

• 如果s不以’-‘开头，且不是数字，如’A123’，返回False
• 如果s以’-'开头，但后面没有数字，或者跟的是0，返回False
• s以’0’开头，但s长度大于1，如’0123’，返回False
• s[1:] 都是数字，返回True，否则返回False

def check(s):
    if s is None or len(s) == 0:
        return False
    if not s[0].isdigit() and s[0] != '-':
        return False
    if s[0] == '-' and (len(s) == 1 or s[1] == '0'):
        return False
    if s[0] == '0' and len(s) != 1:
        return False

    for c in s[1:]:
        if not c.isdigit():
            return False

    return True

若检查失败直接返回0
否则逐个数字识别并计算数值，注意32位整数范围是[0x7fffffff, 0x80000000].

def convert(s):
    if not check(s):
        return 0

    num = 0
    positive, i = (True, 0) if s[0] != '-' else (False, 1)
    overflow = 0x80000000
    for c in s[i:]:
        a = ord(c) - ord('0')
        if (num > overflow//10) or (num == overflow//10 and a > overflow % 10):
            return 0

        num = num*10 + a

    if num == overflow and positive:
        return 0

    return num if positive else -num

测试

def test_convert():
    assert(convert('') == 0)
    assert(convert('-') == 0)
    assert(convert('-0') == 0)
    assert(convert('--') == 0)
    assert(convert('-012') == 0)
    assert(convert('-x') == 0)
    assert(convert('x') == 0)
    assert(convert('123x') == 0)
    assert(convert('0123') == 0)

    assert(convert('0') == 0)
    assert(convert('123') == 123)
    assert(convert('-123') == -123)
    assert(convert('2147483647') == 2147483647)
    assert(convert('2147483648') == 0)
    assert(convert('-2147483648') == -2147483648)

    print('done')


if __name__ == '__main__':
    test_convert()