python LeetCode 两数相除

近一个月一直在写业务,空闲时间刷刷leetcode,刷题过程中遇到了一道比较有意思的题目,和大家分享。

题目描述:

给定两个整数,被除数 dividend 和除数 divisor。将两数相除,要求不使用乘法、除法和 mod 运算符。返回被除数 dividend 除以除数 divisor 得到的商。

示例 1:

输入: dividend = 10, divisor = 3
输出: 3

示例 2:

输入: dividend = 7, divisor = -3
输出: -2

说明:

  • 被除数和除数均为 32 位有符号整数。
  • 除数不为 0。
  • 假设我们的环境只能存储 32 位有符号整数,其数值范围是 [−2**31,  2**31 − 1]。本题中,如果除法结果溢出,则返回 2**31 − 1。

第一反应是这道题还是挺简单的,用减法实现除法不就好了,python刷题实现甚至可以直接使用range()来实现除法,需要注意的点如下:

1.提前判断结果的正负号

2.结果在[-2**31,2**31-1]中,要判断结果是否移除

3.使用range()来计算除法时,一旦除法可以整除我们要对结果+1,因为len(range(3,7,3))的结果是2,len(range(3,9,3))的结果也是2

代码如下:

class Solution(object):
def divide(self, dividend, divisor):
"""
:type dividend: int
:type divisor: int
:rtype: int
"""
below = 1
if dividend < 0 < divisor or divisor < 0 < dividend:
below = -1 dividend, divisor = abs(dividend), abs(divisor)
if dividend < divisor:
return 0
elif divisor == 1:
result = dividend * below
if result >= 2**31-1:
return 2**31-1
return result result = len(range(divisor, dividend, divisor))
if (result+1) * divisor == dividend:
result += 1
return result * below

自己写了好多case来测试都是没问题的,代码提交到leetcode,悲剧了。。。内存错误,看来是内存超了。问题出在核心语句len(range(divisor, dividend, divisor))上,怎么既能保证目前代码的简洁性又能降低内存使用呢。我解决办法是使用xrange代替range,简单的说range返回的对象是个list,会开辟一个很大的空间,而xrange不同,返回的是生成器,所以对内存的使用得到了直接的优化。重新提交,果然通过了。

最终代码如下:

class Solution(object):
def divide(self, dividend, divisor):
"""
:type dividend: int
:type divisor: int
:rtype: int
"""
below = 1
if dividend < 0 < divisor or divisor < 0 < dividend:
below = -1 dividend, divisor = abs(dividend), abs(divisor)
if dividend < divisor:
return 0
elif divisor == 1:
result = dividend * below
if result >= 2**31-1:
return 2**31-1
return result result = len(xrange(divisor, dividend, divisor))
if (result+1) * divisor == dividend:
result += 1
return result * below

希望对大家有所帮助~

上一篇:[LeetCode]29. Divide Two Integers两数相除


下一篇:网站集成QQ登录功能(转)