算法:
有一个整数值
比如,42,它的被除数可以是 1, 2, 3, 6, 7, 14, 21, 42。
它的被除数元素的平方为:1, 4, 9, 36, 49, 196, 441, 1764.
这些平方后的值相加为2500。
而2500,是可被整数平方的,即为整数50的平方。这就是我们要找的值。
若给出整数值在[m,n]区间,我们可以得到这样的序列:
list_squared(1, 250) –> [[1, 1], [42, 2500], [246, 84100]]
list_squared(42, 250) –> [[42, 2500], [246, 84100]]
请写出以上的算法。
fn list_squared(m: u64, n: u64) -> Vec<(u64, u64)> {
let temp: Vec<Vec<u64>> = (m..n + 1)
.map(|x| (1..x + 1).filter(|y| x % y == 0).collect::<Vec<_>>())
.filter(|x| {
let temp: u64 = x.into_iter().map(|y| y.pow(2)).sum();
is_sqrt_ok(temp)
})
.collect();
let tp: Vec<(u64, u64)> = temp.into_iter()
.map(|x| {
let lst = x.last().unwrap();
let s = x.iter().map(|x| x.pow(2)).sum();
(*lst, s)
})
.collect();
tp
}
fn is_sqrt_ok(data: u64) -> bool {
((data as f64).sqrt() as u64).pow(2) == data
}
或简化一下:
fn list_squared(m: u64, n: u64) -> Vec<(u64, u64)> {
(m..n + 1)
.map(|x| (1..x + 1).filter(|y| x % y == 0).collect::<Vec<_>>())
.filter(|x| {
let temp: u64 = x.into_iter().map(|y| y.pow(2)).sum();
is_sqrt_ok(temp)
})
.map(|x| {
let lst = x.last().unwrap();
let s = x.iter().map(|x| x.pow(2)).sum();
(*lst, s)
})
.collect()
}