【acwing算法基础课】 数学知识 快速幂

题目来源：快速幂

题目描述

给定n组ai,bi,pi，对于每组数据，求出ai^bi mod pi的值。

输入格式
第一行包含整数n。

接下来n行，每行包含三个整数ai,bi,pi。

输出格式
对于每组数据，输出一个结果，表示ai^bi mod pi的值。

每个结果占一行。

数据范围
1≤n≤100000,
1≤ai,bi,pi≤2∗109
输入样例：

2
3 2 5
4 3 9

输出样例：

4
1

思路

首先 求a^b mod p，第一反应肯定是暴力

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;


int main()
{
    int n;
    scanf("%d",&n);
    while(n--)
    {
        int a,b,p;
        scanf("%d%d%d",&a,&b,&p);
		int res=1;
		while(b--)
		{
			res=(LL)res*a%p;
		}
        printf("%d\n",res);
    }
    return 0;
}

看着代码很简单 但是由于b可能很大，会超时
那我们就想想如何去优化计算的次数

一个典型的优化方式就是快速幂
首先我们需要知道一些数学知识：

1. (a*b) mod c =((a mod c)*(b mod c)) mod c
2. 一个数转化为2进制的方法

好烦啊 旁边没纸和笔，下次补上！

代码

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;

int qmi(int a,int b,int p)
{
    LL res=1 % p; //p可能是1 然后res就是0
    while(b)
    {
        if(b & 1) res=res*a%p; //末尾是1的话 
        a=(LL)a*a%p;
        b>>=1;
    }
    return res;
}

int main()
{
    int n;
    scanf("%d",&n);
    while(n--)
    {
        int a,b,p;
        scanf("%d%d%d",&a,&b,&p);
        printf("%d\n",qmi(a,b,p));
    }
    return 0;
}