Python 实现大整数乘法算法的示例代码


Posted in Python onSeptember 17, 2019

我们平时接触的长乘法,按位相乘,是一种时间复杂度为 O(n ^ 2) 的算法。今天,我们来介绍一种时间复杂度为 O (n ^ log 3) 的大整数乘法(log 表示以 2 为底的对数)。

介绍原理

karatsuba 算法要求乘数与被乘数要满足以下几个条件,第一,乘数与被乘数的位数相同;第二,乘数与被乘数的位数应为  2 次幂,即为 2 ^ 2,  2 ^ 3, 2 ^ 4, 2 ^ n 等数值。

下面我们先来看几个简单的例子,并以此来了解 karatsuba 算法的使用方法。

两位数相乘

我们设被乘数 A = 85,乘数 B = 41。下面来看我们的操作步骤:

将 A, B 一分为二,令 p = A 的前半部分 = 8,q = A 的后半部分 = 5 , r = B 的前半部分 = 4 ,s = B 的后半部分 =  1,n = 2。通过简单的数学运算:

A * B = pq * rs = (p * 10 + q) * (r * 10 + s)  = p * r * 10 ^ 2 + (p * s + q * r ) * 10 + q * s。

令 u = p * r,v =(p - q) * (s - r),w = q * s。所以 A * B =  u * 10 ^ 2 + (u + v + w) * 10 + w。

换成数值求解的过程如下:

A * B = 85 * 41 = (8 * 10 + 5) * ( 4 * 10 + 1) = 8 * 4 * 10 * 10 + (8 * 1 + 5 * 4) * 10 + 5 * 1。

其中 u = 8 * 4 = 32,v = (8 - 5) (1 - 4) = -9,w = 5 * 1 = 5。

所以,A * B = 32 * 100 + (32 - 9 + 5) * 10 + 5 = 3485。与长乘法所得结果一致。

四位数相乘

我们设被乘数 A = 8537,乘数 B = 4123。下面来看我们的操作步骤:

将 A, B 一分为二,令 p = A 的前半部分 = 85,q = A 的后半部分 = 37 , r = B 的前半部分 = 41 ,s = B 的后半部分 =  23,n = 4。

==> 其中,u = 85 * 41, v = (85 - 37) * (23 - 41), w = 37 * 23。

==> A * B = 8537 * 4123 = u * 10 ^ 4 + (u + v + w) * 10 ^ 2 + w =  3485_0000 +34_7200 + 851 = 35198051。

在我们计算 u, v,  w 的过程中又会涉及两位数的乘法,我们继续使用 Karatsuba 算法得出两位数相乘的结果。

N 位数相乘

我们令 n 为 乘数与被乘数的位数,令 p = A 的前半部分,q = A 的后半部分, r = B 的前半部分 ,s = B 的后半部分。

==> 其中, u = p * r,v = (p - q) * (s - r),w = q * s。

所以 A * B =  u * 10 ^ n + (u + v + w) * 10 ^ (n / 2) + w。

而 u, v, w 则是两个 n / 2 位的乘法运算。我们继续调用 Karatsuba 算法计算 u, v, w 的数值。接着,我们在计算 n / 2 乘法的过程中又会遇到 n / 4 位的乘法运算……以此类推,直到我们遇到两个个位数的乘法,我们就直接返回这两个个位数乘法的结果。层层返回,最终得到 N 位数的乘法结果。

时间复杂度

我们平常使用的长乘法,是 O (n ^ 2) 的时间复杂度。比如两个 N 位数相乘,我们需要将每一位按规则相乘,所以需要计算  N * N 次乘法。而使用  Karatsuba 算法每层需要计算三次乘法,两次加法,以及若干次加法,每使用一次 karatsuba 算法,乘法规模就下降一半。

所以,对于两个 n =  2 ^ K 位数乘法运算,我们需要计算 3 ^ k 次乘法运算。而 K = log n(底数为 2), 3 ^ K = 3 ^ log n = 2  ^ (log 3 * log n) = 2 ^ (log n * log 3) = n ^ log 3 (底数为 2)。

代码实现

from math import log2, ceil
 
def pad(string: str, real_len: int, max_len: int) -> str:
  pad_len: int = max_len - real_len
  return f"{'0' * pad_len}{string}"
 
 
def kara(n1: int, n2: int) -> int:
  if n1 < 10 or n2 < 10:
    return n1 * n2
  n1_str: str = str(n1)
  n2_str: str = str(n2)
  n1_len: int = len(n1_str)
  n2_len: int = len(n2_str)
  real_len: int = max(n1_len, n2_len)
  max_len: int = 2 ** ceil(log2(real_len))
  mid_len: int = max_len >> 1
  n1_pad: str = pad(n1_str, n1_len, max_len)
  n2_pad: str = pad(n2_str, n2_len, max_len)
  p: int = int(n1_pad[:mid_len])
  q: int = int(n1_pad[mid_len:])
  r: int = int(n2_pad[:mid_len])
  s: int = int(n2_pad[mid_len:])
  u: int = kara(p, r)
  v: int = kara(q-p, r-s)
  w: int = kara(q, s)
  return u * 10 ** max_len + (u+v+w) * 10 ** mid_len + w

输出结果:

==> kara(123456, 9734) == 123456 * 9734

==> kara(1234233456756, 32459734) == 1234233456756 * 32459734

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python使用心得之获得github代码库列表
Jun 25 Python
Python操作SQLite简明教程
Jul 10 Python
python中from module import * 的一个坑
Jul 20 Python
将Python中的数据存储到系统本地的简单方法
Apr 11 Python
Python中用PIL库批量给图片加上序号的教程
May 06 Python
关于pip的安装,更新,卸载模块以及使用方法(详解)
May 19 Python
用python编写第一个IDA插件的实例
May 29 Python
python 获取微信好友列表的方法(微信web)
Feb 21 Python
详解django中url路由配置及渲染方式
Feb 25 Python
django模板获取list中指定索引的值方式
May 14 Python
浅谈keras通过model.fit_generator训练模型(节省内存)
Jun 17 Python
详解Python函数print用法
Jun 18 Python
Python对接 xray 和微信实现自动告警
Sep 17 #Python
Python计算两个矩形重合面积代码实例
Sep 16 #Python
详解Python3 pickle模块用法
Sep 16 #Python
python scipy卷积运算的实现方法
Sep 16 #Python
python 三元运算符使用解析
Sep 16 #Python
python 利用pywifi模块实现连接网络破解wifi密码实时监控网络
Sep 16 #Python
Python循环实现n的全排列功能
Sep 16 #Python
You might like
thinkphp实现上一篇与下一篇的方法
2014/12/08 PHP
php精确的统计在线人数的方法
2015/10/21 PHP
Yii输入正确验证码却验证失败的解决方法
2017/06/06 PHP
javascript form 验证函数 弹出对话框形式
2009/06/23 Javascript
在jQuery 1.5中使用deferred对象的代码(翻译)
2011/03/10 Javascript
20个最新的jQuery插件
2012/01/13 Javascript
由点击页面其它地方隐藏div所想到的jQuery的delegate
2013/08/29 Javascript
上传文件返回的json数据会被提示下载问题解决方案
2014/12/03 Javascript
js判断浏览器类型及设备(移动页面开发)
2015/07/30 Javascript
JS中递归函数
2016/06/17 Javascript
JavaScript面向对象分层思维全面解析
2016/11/22 Javascript
JS实现的简单表单验证功能示例
2017/10/13 Javascript
Vue如何从1.0迁移到2.0
2017/10/19 Javascript
详解vuex的简单todolist例子
2019/07/14 Javascript
JavaScript使用localStorage存储数据
2019/09/25 Javascript
layui下拉列表select实现可输入查找的方法
2019/09/28 Javascript
[02:27]《DAC最前线》之附加赛征程
2015/01/29 DOTA
django自定义Field实现一个字段存储以逗号分隔的字符串
2014/04/27 Python
python实现挑选出来100以内的质数
2015/03/24 Python
深入理解Python3中的http.client模块
2017/03/29 Python
Python基于OpenCV实现视频的人脸检测
2018/01/23 Python
python3 中文乱码与默认编码格式设定方法
2018/10/31 Python
500行Python代码打造刷脸考勤系统
2019/06/03 Python
一篇文章了解Python中常见的序列化操作
2019/06/20 Python
python返回数组的索引实例
2019/11/28 Python
python 操作hive pyhs2方式
2019/12/21 Python
PyInstaller将Python文件打包为exe后如何反编译(破解源码)以及防止反编译
2020/04/15 Python
python 实现表情识别
2020/11/21 Python
新闻专业推荐信范文
2013/11/20 职场文书
大学生职业生涯规划范文
2014/01/08 职场文书
摄影专业毕业生求职信
2014/03/13 职场文书
基层工作经历证明
2015/06/19 职场文书
服务行业标语口号
2015/12/26 职场文书
Django migrate报错的解决方案
2021/05/20 Python
AJAX引擎原理以及XmlHttpRequest对象的axios、fetch区别详解
2022/04/09 Javascript
Python sklearn分类决策树方法详解
2022/09/23 Python