prime

素数判定算法，经典的Rabin Miller测试，通过二次探测的方法，可以将其正确率上升到一个很高的高度。

$O(1)$的快速乘。

在一些卡常数而且爆long long的取余问题中用到快速乘。

朴素的快速乘是$O(logn)$的，从而添加了不必要的复杂度。

爆long long的，实质上是取余的结果，在long long运算中只要不涉及除法，那么一直是对INF取余的结果，对答案没有干扰。

1 LL mul(LL a,LL b,LL mod){
2     if(a<=(LL)(1e8) && b<=(LL)(1e8)) return a*b%mod;
3     return (a*b - (LL)(a/(LD)mod*b + 1e-3)*mod + mod) % mod;
4 }

注意快速乘在很小的数字时并不太稳，所以特判一下。

注意一下Linux下rand()函数足够大，不需要拓充。

转入正题：Rabin测试。

我的Rabin在INT范围内不会出错，在LOGN LONG范围内出错的概率极低。

首先是写一个check(LL x,LL P)，用数字x检验P是否为素数。

 1 bool check(LL x,LL P){
 2     LL tmp=P-1;
 3     while(!(tmp&1)) tmp>>=1;
 4     LL m=qpow(x,tmp,P);
 5     if(m==1) return 1;
 6     while(tmp<P){
 7         if(m==P-1) return 1;
 8         tmp<<=1; m=mul(m,m,P);
 9     }
10     return 0;
11 }

梳理一下过程

首先将P-1中所有的2全都除掉。

这时候如果tmp 为 P-1 或者 1是伪素数。（具体就用 $x^{tmp} = 1 (mod P)$判定）

这时候检查$x^{2tmp} , x^{4tmp}, x^{8tmp} ....$是否等于 $P-1$。

如果没有则说明P不是素数。

接下来基于这一个$O(logn)$判定素数，我们有$O(n^{1/4})$的分解质因数算法。

Rho算法。

Rho算法是基于一个定理对于一个小于n的数字x如果 $(x,n) ≠ 1$ 则有，(x,n)为n的因子。

这样考虑怎样高效地得到这样的数字x。

 1 LL Rdo(LL n,LL c){
 2     LL i=1,k=2,x,y,d,p;
 3     x=Rand()%n;
 4     y=x;
 5     while(1){
 6         i++;
 7         x=(mul(x,x,n)+c)%n;
 8         if(y==x) return n;
 9         p=Abs(x-y);
10         d=gcd(p,n);
11         if(d!=1&&d!=n) return d;
12         if(i==k){
13             y=x;
14             k+=k;
15         }
16     }
17 }
18 
19 int tot;
20 LL a[N];
21 
22 void down(LL n){
23     if(n==1) return;
24     if(isprime(n)){
25         a[++tot]=n;
26         return;
27     }
28     LL t=n;
29     while(t==n) t=Rdo(n,Rand()%(n-1)+1);
30     down(t);
31     down(n/t);
32 }

这样就可以得到了分解质因数的高效方法。

注意一个数字的质因子最多有$O(logn)$个，因为质因数的乘积为n，而且质因数都大于1.

注意在随机数据情况下，采用朴素分解质因数的效率也大概接近 $O(n^{frac{1}{4}})$，只不过会受到空间的限制。

for(int i=1;prime[i]*(LL)prime[i]<=tmp;i++)
{
    int cnt = 0;
    tim_cnt++;
    while(tmp % prime[i] == 0) tmp /= prime[i];
    ans *= (cnt+1);
}

注意是 <=tmp 而不是 <=n

给定一个整数N，求N最少可以拆成多少个完全平方数的和。

所以根据拉格朗日平方和定理，答案为1~4

答案为1，开方验证

答案为2，根据勾股数定理一个数字为勾股数当且仅当其质因数分解中所有的$4n+3$项的指数为偶数。

答案为3，根据初等数论中的不等式 $n ≠ (8k + 7) cdot 4^{m}$

不然答案为4，根据拉格朗日平方和定理即可。