快速幂-java

程序猿零零漆

已于 2024-03-24 19:53:08 修改

阅读量1k

点赞数 24

文章标签： java

于 2024-03-24 00:01:08 首次发布

本文链接：https://blog.csdn.net/m0_63267251/article/details/136976988

版权

本文介绍了快速幂算法，通过将常规的O(n)时间复杂度优化为O(logn)，并通过实例展示了如何在Java中实现，并扩展到包含取模运算的版本。通过二进制表示法，理解并计算幂的高效方法。

摘要由CSDN通过智能技术生成

文章目录

前言

主要将求 $a^k$ 的值将时间复杂度为O(n)的算法优化为O( $logn$ )

一、快速幂是什么？

通常我们如果要求 $a^k$ 的值，往往是通过一个循环k次，然后将res初始化为1，让res = res * a,然而当数据过于庞大时，这种算法往往会超时，那么我们就可以通过快速幂算法将时间复杂度优化为O( $logn$ )。

二、算法思路

1.1 比如求 $4^5$ 值

将5转化成二进制数101，然后将4的5次方拆分为 $4^1$ * $4^4$ = $4^5$ ，显而易见， $4^1$ 中的1即为 $2^0$ , $4^4$ 中的4即为 $2^2$ 。（注：5的二进制数从最右往左为 $1*2^0+0*2^1+1*2^2$ = 5）

同时我们发现最终拆分时，拆成的两项分别是二进制数中对应为1的位置，那么 $4^1$ * $4^4$ 其实可以写成

其实中间还有一位是4^2^1，但是它对应的二进制数为0，故不参与计算。

而且其实后一位数是前一位数的平方，那么我们可以很明显的得到规律

三、使用步骤

1.代码如下（示例）：


import java.io.*;

public class 快速幂 {
    //java快读快写的模板
    static PrintWriter pw = new PrintWriter(new OutputStreamWriter(System.out));
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    static StreamTokenizer st = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));

    public static void main(String[] args) throws Exception{
        //输入多个样例数
        int n = nextInt();
        while(n-- > 0) {
            //读入存入的a
            long a = nextLong();
            //读入存入的k，即为a的次方数
            long k = nextLong();
            //初始化结果result默认值为1
            long result = 1;
            //当k为0是退出循环，相当于k的二进制数我们已经遍历结束
            while (k != 0) {
                //判断k的二进制数最后一位是否为1
                if (k % 2 == 1) {
                    //此时我们将result * a
                    result = result * a ;
                }
                //相当于二进制数右移，即去掉已经被我们判断过的二进制最后一位数
                k /= 2;
                //将此时的a进行平方运算，例：a^2^0 变为a^2^1
                a = a * a ;
            }
            //打印最后的结果result
            pw.println(result);
        }
        pw.flush();
    }

    public static int nextInt()throws Exception{
        st.nextToken();
        return (int)st.nval;
    }
    public static long nextLong()throws Exception{
        st.nextToken();
        return (long)st.nval;
    }

}

2.读入数据

读入测试样例（示例）：

得到的代码运行结果

243
256
32

四、扩展

通常考题会再加一个数字p，让我们来对最后的结果进行取模运算，那么我们也可以将上述示例中的 $4^1$ * $4^4$ = $4^5$ 中的 $4^1$ 进行取模运算和 $4^4$ 进行取模运算，将两者得到的结果相乘最后进行一次取模运算，就可以得到最后的结果

1.代码如下（示例）：


import java.io.*;

public class 快速幂 {
    //java快读快写的模板
    static PrintWriter pw = new PrintWriter(new OutputStreamWriter(System.out));
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    static StreamTokenizer st = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));

    public static void main(String[] args) throws Exception{
        //输入多个样例数
        int n = nextInt();
        while(n-- > 0) {
            //读入存入的a
            long a = nextLong();
            //读入存入的k，即为a的次方数
            long k = nextLong();
            //对所求的结果mod p（即对p进行取模运算）
            long p =nextLong();
            //初始化结果result默认值为1
            long result = 1;
            //当k为0是退出循环，相当于k的二进制数我们已经遍历结束
            while (k != 0) {
                //判断k的二进制数最后一位是否为1
                if (k % 2 == 1) {
                    //此时我们将result * a后再对p进行取模运算
                    result = result * a % p;
                }
                //相当于二进制数右移，即去掉已经被我们判断过的二进制最后一位数
                k /= 2;
                //将此时的a进行平方运算再进行取模运算
                a = a * a % p;
            }
            //打印最后的结果result对p进行取模运算
            pw.println(result%p);
        }
        pw.flush();
    }

    public static int nextInt()throws Exception{
        st.nextToken();
        return (int)st.nval;
    }
    public static long nextLong()throws Exception{
        st.nextToken();
        return (long)st.nval;
    }

}