牛客竞赛——a^b

题目描述

求 a 的 b 次方对 p 取模的值，其中 0≤a,b,p≤10^9​

 

输入描述:

三个用空格隔开的整数a,b和p。

输出描述:

一个整数，表示 a^b mod p 的值。

示例1

输入

2 3 9

输出

8

 

分析

这题其实就是用快速幂写，不过就是要注意以下n == 0且p == 1时的情况，需要特判以下，因为任何非负数 mod 1都为0，如果不特判最后就会返回 1，与实际不符

 

AC代码

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        int a = input.nextInt();
        int b = input.nextInt();
        int p = input.nextInt();
        long res = qpow(a, b, p);
        System.out.println(res);
    }
    
    public static long qpow(long a, long n, long mod) {
        if (a == 0) {
            return 0;
        }
        
        // 任何一个非负数对1求模，都是0，因为1都是他们的公约数
        // 这里需要他特判一下
        if (n == 0) {
            if (mod != 1) {
                return 1;
            }
            return 0;
        }
        
         long ans = 1;
         while (n != 0) {
             if (n % 2 == 1) {
                 ans *= a;
                 ans %= mod;
             }
             a *= a;
             a %= mod;
             n = n >> 1;
         }
        
        return ans;
    }
}

牛客链接：https://ac.nowcoder.com/acm/contest/996/A