C++编程提升之路第二课——数学

一：前言

众所周知，数学在我们编程当中是非常重要的。一些题目，你用纯暴力肯定会超时，而经过数学优化后复杂度就会降低很多。

二：欧几里得算法

1：介绍

• 其实这个算法就是 $gcd(a,b)=gcd(b,amodb)$ 但是要证明它并不是很容易。

• 我们可以先证明： $gcd(a,b)=gcd(b,a-b)$

  • 我们设 $d=gcd(a,b)$ ,则 $a=d\cdot a1$ ,$b=d\cdot b1$ 且 $gcd(a1,b1)=1$

  • 且 $a-b=d\cdot esa1-d\cdot b1=d\cdot (a1-b1)$

  • 那么 $a-b$ 和 $b$ 都有一个公因子 $d$，接下来只需证明：

  • $gcd(b1,a1-b1)=1$

  • 我们采用反证法

  • 设 $b1=\beta \cdot \lambda$, $a1-b1=\alpha \cdot \lambda (\lambda >1)$

  • 则 $a1=a1-b1+b1=\lambda \cdot (\alpha +\beta )$

  • 因为之前我们已经定义$gcd(a1,b1)=1$且$b1=\lambda \cdot \beta$

  • 那么$\lambda$只能等于1

  • 与命题不符，得证

• $\because gcd(a,b)=gcd(b,a-b)$

• $\therefore gcd(a,b)=gcd(b,a-b)=gcd(b,a-2b)=gcd(b,a-3b)...$

• $\therefore gcd(a,b)=gcd(b,amodb)$

• 得证！

• 根据此定理，我们可以快速求出两个数的 $gcd$ 。

2：code

• 代码：

#include<bits/stdc++.h>
using namespace std;
inline int gcd(int n,int m){
	if(b==0)return a;
	return gcd(b,a%b);
}
int main(){
	int n,m;
	cin>>n>>m;
	cout<<gcd(n,m);
	return 0;
}

三：扩展欧几里得算法

1:介绍

• 它就是求 $ax+by=gcd(a,b)$
  • 显然，当 $b=0$ 时，则原方程变为 $ax=a$,解为$\{\begin{array}{l}x=1\\ y=0\end{array}$
  • 若 $b\ne 0$，且有 $x_1,y_1$满足$amodb\cdot x_1+b\cdot y1=gcd(b,amodb)$，则有解 $\{\begin{array}{l}x=x_1\\ y=y_1-x_1\cdot floor(a/b)\end{array}$
  • 下面为证明过程
    • $\because (amodb)\cdot x_1+b\cdot y_1=gcd(b,amodb)$
    • $\therefore (a-floor(a/b)\cdot b)\cdot x_1+b\cdot y_1=gcd(a,b)$
    • $\therefore a\cdot x_1+b\cdot (y_1-x_1\cdot (floor(a/b))=gcd(a,b)$
    • 证毕
  • 根据此结论，我们可以一直递归，直至 $b=0$

2：code

• 代码：

#include<bits/stdc++.h>
using namespace std;
void ex_gcd(int a,int b,int &x,int &y){//求a*x+b*y=gcd(a,b)的整数解 
	if(!b){
		x=1;
		y=0;
		return;
	}
	ex_gcd(b,a%b,y,x);
	y-=(a/b)*x;
}
int main(){
	int a,b,x,y;
	cin>>a>>b;
	ex_gcd(a,b,x,y);
	cout<<x<<' '<<y;
	return 0;
}

3：拓展1

• 题目：解方程 $ax+by=c$
  • 思路：我们设 $d=gcd(a,b)$ ,再把等式两边同时乘上$\frac{d}{c}$
  • 则原方程化为$a\cdot \frac{d}{c}\cdot x+b\cdot \frac{d}{c}\cdot y=gcd(a,b)$
  • 求解一个$a\cdot x^{\prime }+b\cdot y^{\prime }=gcd(a,b)$
  • 则方程的解为$\{\begin{array}{l}x=x^{\prime }\cdot \frac{d}{c}\\ y=y^{\prime }\cdot \frac{d}{c}\end{array}$
  • 若$c\nmid d$,则原方程无解

4：拓展2

• 题目：解方程 $a\cdot x\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$
  • 思路：此题非常简单，将它化为拓展1
  • 原方程化为 $ax+py=b$ 再用拓展1解即可
  • 若 $b=1$ ,则此题化为了求解 $a$ 在 $modp$ 意义下的逆元。

四：逆元的求法

• 逆元的定义为若 $a\cdot x\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$
• 求解方式有一下几种：

1：费马小定理（p为质数）

• 费马小定理：$a^{p-1}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}p)$,证明此处省略
• 则$a\cdot a^{p-2}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}p)$
• $\therefore a$在$modp$意义下的逆元为$a^{p-2}$
• 只需使用一个快速幂就好了
• 代码：

#include<bits/stdc++.h>
using namespace std;
int power(int a,int b,int p){
	int ans=1;
	while(b){
		if(b&1)ans=ans*a%p;
		a=a*a%p;b>>=1;
	}
	return ans;
}
int inv(int x,int p){
	return power(x,p-2,p);
}
int main(){
	int x,p;
	cin>>x>>p;
	cout<<inv(x,p);
	return 0;
}

2：线性求逆元（p为质数）

• 当我们要求 $1$到$n$ 中所有的数的逆元
• 我们用一个 $inv[i]$ 来存 $i$ 的逆元
• 设我们要求的是 $inv[i]$ ,且 $p=k\cdot i+r(r<i)$
• 又因为 $p$ 为质数，则 $0<r<i$
• 且 $k=\lfloor x\rfloor ,r=pmodi,k\times i+r\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}p)$
• 将等式两边同时乘以 $i,r$ 的逆元
• $k\cdot inv[r]+inv[i]\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}p)\Rightarrow inv[i]\equiv -k\cdot inv[r](\mathrm{m}\mathrm{o}\mathrm{d}p)d\Rightarrow inv[i]\equiv \lfloor \frac{p}{i}\rfloor \cdot inv[pmodi](\mathrm{m}\mathrm{o}\mathrm{d}p)$
• $\therefore inv[i]=(p-\lfloor \frac{p}{i}\rfloor \cdot inv[pmodi])$
• 根据此递推式，即可写出代码

#include<bits/stdc++.h>
using namespace std;
const int N=3e6+7;
int n,p;
long long inv[N];
int main(){
	cin>>n>>p;
	inv[1]=1;
	for(int i=2;i<=n;i++)
	inv[i]=(p-p/i)*inv[p%i]%p;
	for(int i=1;i<=n;i++){
		cout<<inv[i]<<"\n";
	}
	return 0;
}

3:扩展欧几里得求逆元

• 根据我们之前学的扩展欧几里得拓展2的例题：
• 求$a\cdot x\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$
• 将b换为1即可

#include<bits/stdc++.h>
using namespace std;
void ex_gcd(int a,int b,int &x,int &y){//求a*x+b*y=gcd(a,b)的整数解 
	if(!b){
		x=1;
		y=0;
		return;
	}
	ex_gcd(b,a%b,y,x);
	y-=(a/b)*x;
}
int get_inv(int a,int p){
	int x,y;
	ex_gcd(a,p,x,y);
	x=(x%p+p)%p;//求最小的逆元
	return x; 
} 
int main(){
	int a,p;
	cin>>a>>p;
	cout<<get_inv(a,p);;
	return 0;
}

5:结语

想必通过今天作者对大家的总结，大家的数学水平都应该有一定的提升了吧，写code的能力估计也强了好几倍，希望在这段时间了，我们一起努力，一起进步！有什么错误还请各位大佬指出。