Optimize calculations

master
trivernis 5 years ago
parent b6cd114e6b
commit b87c359239

@ -1,6 +1,6 @@
use num_bigint::{BigUint, ToBigUint}; use num_bigint::{BigUint, ToBigUint};
use num_integer::Integer; use num_integer::Integer;
use num_traits::ToPrimitive; use num_traits::{ToPrimitive, Zero};
use std::sync::mpsc::*; use std::sync::mpsc::*;
use std::ops::MulAssign; use std::ops::MulAssign;
use std::thread; use std::thread;
@ -10,7 +10,7 @@ fn main() {
let num_threads: u64 = num_cpus::get() as u64; let num_threads: u64 = num_cpus::get() as u64;
let (tx, rx) = channel::<BigUint>(); let (tx, rx) = channel::<BigUint>();
let (sen_timing, rec_timing) = channel::<bool>(); let (sen_timing, rec_timing) = channel::<bool>();
let end: BigUint = ubig_pow(10.to_biguint().unwrap(), 100); let end: BigUint = ubig_pow(10, 100);
println!("Starting {} threads...", num_threads); println!("Starting {} threads...", num_threads);
for i in 0u64..num_threads { for i in 0u64..num_threads {
let tx = tx.clone(); let tx = tx.clone();
@ -47,46 +47,51 @@ fn main() {
fn get_rotatable(tx: Sender<BigUint>, sen_time: Sender<bool>, start: u64, end: BigUint, step: u64) { fn get_rotatable(tx: Sender<BigUint>, sen_time: Sender<bool>, start: u64, end: BigUint, step: u64) {
let mut num: BigUint = start.to_biguint().unwrap(); let mut num: BigUint = start.to_biguint().unwrap();
let zero: BigUint = 0.to_biguint().unwrap(); let zero: BigUint = 0.to_biguint().unwrap();
let ten: BigUint = 10.to_biguint().unwrap();
while num < end { while num < end {
if num.is_odd() || !num.is_multiple_of(&ten) {
let mut digits = ubig_digits(num.clone()); let mut digits = ubig_digits(num.clone());
digits.rotate_right(1); if digits.first() >= digits.last() {
digits.rotate_left(1);
let num_rotated = ubig_from_digits(digits); let num_rotated = ubig_from_digits(digits);
if (num_rotated % &num) == zero { if (num_rotated % &num).is_zero() {
let _ = tx.send(num.clone()); let _ = tx.send(num.clone());
} }
}
}
num += step; num += step;
let _ = sen_time.send(true); let _ = sen_time.send(true);
} }
} }
/// returns a vector containing the digits of the BigUint /// returns a vector containing the digits of the BigUint
fn ubig_digits(big_number: BigUint) -> Vec<u32> { fn ubig_digits(big_number: BigUint) -> Vec<u8> {
let mut num: BigUint = big_number; let mut num: BigUint = big_number;
let mut digits: Vec<u32> = vec![]; let mut digits: Vec<u8> = vec![];
let zero: BigUint = 0.to_biguint().unwrap(); let zero: BigUint = 0.to_biguint().unwrap();
let ten: BigUint = 10.to_biguint().unwrap(); let ten: BigUint = 10.to_biguint().unwrap();
while num > zero { while num > zero {
let (quot, rem) = num.div_rem(&ten); let (quot, rem) = num.div_rem(&ten);
num = quot; num = quot;
digits.push(rem.to_u32().unwrap()); digits.push(rem.to_u8().unwrap());
} }
digits digits
} }
/// returns a BigUint for a vector of digits /// returns a BigUint for a vector of digits
fn ubig_from_digits(digits: Vec<u32>) -> BigUint { fn ubig_from_digits(digits: Vec<u8>) -> BigUint {
let mut num: BigUint = 0.to_biguint().unwrap(); let mut num: BigUint = 0.to_biguint().unwrap();
for (index, digit) in digits.iter().enumerate() { for (index, digit) in digits.iter().enumerate() {
num += digit * (10 as u32).pow(index as u32); num += ubig_pow(10, index) * digit;
} }
num num
} }
/// returns ubig^exp /// returns ubig^exp
fn ubig_pow(ubig: BigUint, exp: usize) -> BigUint { fn ubig_pow(base: u128, exp: usize) -> BigUint {
let mut num = ubig.clone(); let mut num = base.to_biguint().unwrap();
for _ in 0..exp { for _ in 0..exp {
num.mul_assign(ubig.clone()) num.mul_assign(base)
} }
num num
} }
Loading…
Cancel
Save