试题 算法提高 JOE的算数(快速幂运算)

JOE的算数

题目

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　有一天，JOE终于不能忍受计算a^{b%c这种平凡的运算了。所以他决定要求你写一个程序，计算a}b%c。

提示：若b为奇数,，a^b=(a(b/2))^{2*a，否则a}b=(a^(b/2))2。
输入格式
　　三个非负整数a,b,c；
输出格式
　　一个整数ans，表示a^b%c；
样例输入
7 2 5
样例输出
4
数据规模和约定
　　30% a <= 100, b <= 10^4, 1 <= c <= 100
　　60% a <=10^4, b <= 10^5, 1 <= c <= 10^4
　　100% a <=10^6, b <= 10^9, 1 <= c <= 10^6

解法

需要注意数据规模,由于幂运算的指数非常大,如果用迭代法会超时,所以需要运用快速幂运算,同时在计算过程中不断进行模运算.

import java.util.Scanner;

public class Main {
//注意精度  用大数不然数据溢出
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		long a = sc.nextLong();
		long b = sc.nextLong();
		long c = sc.nextLong();
		System.out.println(method1(a, b, c));
	}
	//快速幂运算
	static long method1(long a,long b,long c) {
		long pf=a;//a的1次方
		long res=1;
		long m=b/2;
		if(b!=0) {
			while(m!=0) {
				if((m&1)==1) {
					res*=pf;
					res%=c;
				}
				pf*=pf;
				pf%=c;
				m>>=1;
			}
		}else {
			res=1;
		}
		if(b%2==0) {
			return res*res%c;
		}else {
			return res*res*a%c;
		}
	}
}