RosettaCodeData/Task/List-comprehensions/Rust/list-comprehensions-1.rust
2023-07-01 13:44:08 -04:00

13 lines
354 B
Text

fn pyth(n: u32) -> impl Iterator<Item = [u32; 3]> {
(1..=n).flat_map(move |x| {
(x..=n).flat_map(move |y| {
(y..=n).filter_map(move |z| {
if x.pow(2) + y.pow(2) == z.pow(2) {
Some([x, y, z])
} else {
None
}
})
})
})
}