反转整数
给定一个 32 位有符号整数,将整数中的数字进行反转。
示例 1:
输入: 123
输出: 321
示例 2:
输入: -123
输出: -321
示例 3:
输入: 120
输出: 21
注意:
假设我们的环境只能存储 32 位有符号整数,其数值范围是 [ $−2^{31}$, $2^{31}$ − 1 ]。根据这个假设,如果反转后的整数溢出,则返回 0。
In [1]:
class Solution:
def reverse(self, x):
"""
:type x: int
:rtype: int
"""
flag = 1
if x < 0:
flag = -1
a = int(str(abs(x))[::-1]) * flag
if abs(a) >= 2**31:
return 0
return a
优化¶
当然这里存在几个问题,我直接是直接使用了字符串将其绝对值反转, 另外,在最大值上其实也有问题;
假设代码运行只支持32位整形,那么实际上这段代码是不可执行的,因为在反转之后强转为 int 的时候就会溢出,这边实际上应该考虑这个情况。
这里修改一下代码:
In [2]:
class Solution:
def reverse(self, x):
"""
:type x: int
:rtype: int
"""
int_max = 2147483647 # 2**31 - 1
r = 0
flag = 1
if x < 0:
flag = -1
x = abs(x)
while x != 0:
pop = x % 10 # 倒序,取个位数上的值
x = x // 10 # 整除 10
if r > int_max/10 or (r == int_max/10 and pop > 7):
return 0
r = r * 10 + pop
return r * flag
这里需要在计算 r = r * 10 + pop 之前就判断,是否有溢出的可能, 满足以下任意条件就会溢出。
- 当前
r值大于int_max / 10那么最后一位不管是什么都溢出
- 当前
r值等于int_max / 10但最后一位是大于7,则超过最大数字溢出
Comments !