背景:
暴力求 n 的约数个数、约数和的复杂度为 ,往往不能符合题目时限要求。但这个通过质因数分解来优化时间。
质因数分解:
性质:一个数 n ,最多只存在一个质因子 。
因此,我们只需要预处理出 以内的质数(1e5以内质数大概4000+个)。遍历这些质数,若n%p==0,那么n/p,直到n%p!=0。最后如果 n 还大于 1 ,那么这个剩下的 n 一定也是质数。
设 ,
以内质数个数为 T,则质因数分解复杂度为
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAX = 5e4 + 10;
const int mod = 1e9 + 7;
int n;
int prime[MAX];
int vis[MAX];
int num;
pair<int, int>pr[20];//first为质因子,second为该质因子的个数
void get_prime()
{
num = 0;
for (int i = 2; i <= MAX; i++) {
if (!vis[i]) {
prime[++num] = i;
}
for (int j = 1; j <= num && prime[j] * i <= MAX; j++) {
vis[i*prime[j]] = 1;
if (i%prime[j] == 0) break;
}
}
}
int main()
{
get_prime();
scanf("%d", &n);
int cnt = 0;
for (int i = 1; i <= num; i++) {
if (n%prime[i] == 0) {
pr[cnt] = make_pair(prime[i], 0);
while (n%prime[i] == 0) {
n /= prime[i];
pr[cnt].second++;
}
cnt++;
}
if (n == 1) break;
}
if (n != 1) {
pr[cnt] = make_pair(n, 1);
cnt++;
}
//test
for (int i = 0; i < cnt; i++) {
printf("%d^%d\n", pr[i].first, pr[i].second);
}
return 0;
}
约数个数:
设 ,约数个数 =
。
约数和:
设 。
约数和
(每项都为等比数列求和)
其实也可以每项暴力求和,可以边质因数分解边计算,复杂度甚至更优。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAX = 5e4 + 10;
const int mod = 1e9 + 7;
int n;
int prime[MAX];
int vis[MAX];
int num;
void get_prime()
{
num = 0;
for (int i = 2; i <= MAX; i++) {
if (!vis[i]) {
prime[++num] = i;
}
for (int j = 1; j <= num && prime[j] * i <= MAX; j++) {
vis[i*prime[j]] = 1;
if (i%prime[j] == 0) break;
}
}
}
int main()
{
get_prime();
scanf("%d", &n);
int ans = 1;
int tot = 1;
for (int i = 1; i <= num; i++) {
if (n%prime[i] == 0) {
int w = 0;
ll tmp = 1;
ll sum = 1;
while (n%prime[i] == 0) {
n /= prime[i];
tmp *= prime[i];
sum += tmp;
w++;
}
tot *= (w + 1);
ans = 1ll * sum*ans%mod;
}
if (n == 1) break;
}
if (n != 1) {
tot *= 2;
ans = 1ll * (n + 1)*ans%mod;
}
printf("约数个数:%d\n", tot);
printf("约数和(mod 1e9+7):%d\n", ans);
return 0;
}