Find square root of U256 or any integers

square_root.rs

use alloy::primitives::U256;


fn integer_sqrt(n: U256) -> U256 {
    // https://en.wikipedia.org/wiki/Integer_square_root
    if n < U256::from(2) {
        return n;
    }

    let small_candidate = integer_sqrt(n >> 2) << 1;

    let large_candidate = small_candidate + U256::from(1);

    if large_candidate * large_candidate > n {
        small_candidate
    } else {
        large_candidate
    }
}


fn main() {
    let res = U256::from(64);
    println!("{}", integer_sqrt(res));
}