注意事项:
代码中涉及分解质因数的算法,可以看我之前的文章:java—试除法分解质因数
秦九韶算法,这个就没什么好说的了吧
题目:
给定 n 个正整数 ai,请你输出这些数的乘积的约数之和,答案对 1e9+7 取模
第一行包含整数 n
接下来 n 行,每行包含一个整数 ai
输出一个整数,表示所给正整数的乘积的约数之和,答案需对 1e9+7 取模
输入:
3
2
6
8
输出:
252
public class 约数_约数之和 {
public static int mod = 1000000007;
public static HashMap<Integer, Integer> map = new HashMap<>();
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int m = in.nextInt();
while (m-- > 0) {
int x = in.nextInt();
//这个地方用的是分解质因数的算法,稍微有所不同就是那个算法中我们每次都将出现次数输出,而这里我们是把出现次数存在map中
//拿到x的质因数,并把每个质因数加到map的value值中,例如8的质因数是2*2*2,那么map中2的value值就会总共加3
for (int i = 2; i<=x/i; i++) {
while (x % i == 0) {
map.put(i, map.getOrDefault(i, 0) + 1); //这一句用getOrDefault是防止NullPointerException
x /= i;
}
}
if (x > 1) map.put(x, map.getOrDefault(x, 0) + 1);
}
//res为最终结果,遍历map,对每个相同质因子的不同平方相加,用的是秦九韶算法
//根据题目要求每次计算进行mod处理
long res = 1;
for (Map.Entry<Integer, Integer> i : map.entrySet()) {
long t = 1;
int a = i.getKey(), b = i.getValue(); //这里a是底数,b是次方,例如2^4, a=2,b=4
while (b-- > 0) {
t = (t * a + 1) % mod;
}
res = res * t % mod;
}
System.out.println(res);
}
}
就拿题目举例:
2*6*8=96, 分解之后为(2^5)*(3^1)
然后对每一个进行分解计算
(2^0+2^1+2^2+2^3+2^4+2^5)*(3^0+3^1)=252
声明:算法思路来源为y总,详细请见https://www.acwing.com/
本文仅用作学习记录和交流