数据结构与算法：快速幂——求幂运算 O(logN)

Game_小志

已于 2022-05-04 20:55:39 修改

阅读量1.6k

点赞数 2

分类专栏：数据结构与算法文章标签： c# 算法数据结构

于 2022-05-04 20:48:45 首次发布

本文链接：https://blog.csdn.net/Sea3752/article/details/124576488

版权

数据结构与算法专栏收录该内容

6 篇文章 0 订阅

订阅专栏

前言：普通的求幂问题，相信大家已经屡见不鲜，无非就是 纯暴力 解法，或者直接调用公式：Math.Pow(底数, 指数) 一步得出。但这两种算法只能使得时间复杂度保持在 O(n)。而且公式一步的得往往对求模问题解决不了。对最大值得限制和对时间复杂度的限制，有必要让你学习一下快速幂算法。

题目：求 m 的 n 次方，对 998244353 求余。要求时间复杂度 O(logN)。

常见方法：

1.暴力求
时间复杂度 : O(n) 。

2.数学公式
Math.Pow(m, n)；(对于求模往往会出问题）。

时间复杂度 : O(n) 。

3.快速幂算法
时间复杂度为: O(log n)。

讲解：利用公式 $m^{n} = (m^{2} )^{\frac{n}{2}}$ ,可以直接用底数的平方，指数除以2的方式，这样指数降到0只需要 log(n) 次，因此时间复杂度仅需要 O(log n) 比朴素的模拟乘方法 O(n) 优秀的多。

测试代码（c#语言）：

递归方法：

  static long Recursion(long m,long n,long p)
        {
            if (n == 0) return 1;  
            if (n == 1) return m;
            long part = Recursion(m, n / 2, p);  //递归到最小，然后以此扩大到原数。 
            if(n%2==0)    //判断是否n为奇数
            {
                return part * part % p;
            }
            else
            {
                return m * part * part % p;
            }
        }

对递归函数举例解释：例如 m = 3,n=5;

第一次进入函数： n = 5，

第二次进入函数： n = 2 ( 5/2 = 2)

第三次进入函数： n = 1 ( 2/2 = 1) 返回m ，即返回 3，

使得第二次进入函数时 part 为 3，此时 n = 2 为偶数，返回 part * part 即 3*3= 9；

使得第一次进入函数时 part 为 9，此时 n = 5 为奇数，返回 m*part*part 即 3*9*9 得243

非递归方法：

 static long UnRecursion(long m, long n,long p)
        {
            long res = 1;
            while(n>=1)
            {
                if (n % 2 == 1) res = res * m % p;
                n /= 2;
                m = m * m % p;
            }
            return res;
        }

两种方法具体选择看自身喜好,时间复杂度均为 logn

主测试代码：

 static void Main()
        {
            //求 m 的 n 次方，对 998244353 取余。 
            Console.WriteLine("题目：求m的n次方，对998244353求余。要求时间复杂度O(logN)。");
            Console.WriteLine("请属于m和n：");
            string[] s = Console.ReadLine().Split(' ');
            long m = Convert.ToInt64(s[0]);
            long n = Convert.ToInt64(s[1]);
            long p = 998244353;
            //直接求幂：时间复杂度O(n):
            Console.WriteLine("运用求幂公式直接得(超过最大值报错)："+Math.Pow(m, n)%p);

            Console.WriteLine("                      调用递归结果：" + Recursion(m, n, p));

            Console.WriteLine("                调用非递归递归结果：" + UnRecursion(m, n, p));

            Console.Read();
        }

测试结果截图：

3的5次方测试：