Integer square root
In number theory, the integer square root (isqrt) of a non-negative integer n is the non-negative integer m which is the greatest integer less than or equal to the square root of n,
For example,
Introductory remark
[edit]Let and be non-negative integers.
Algorithms that compute (the decimal representation of) run forever on each input which is not a perfect square.[note 1]
Algorithms that compute do not run forever. They are nevertheless capable of computing up to any desired accuracy .
Choose any and compute .
For example (setting ):
Compare the results with
It appears that the multiplication of the input by gives an accuracy of k decimal digits.[note 2]
To compute the (entire) decimal representation of , one can execute an infinite number of times, increasing by a factor at each pass.
Assume that in the next program () the procedure is already defined and — for the sake of the argument — that all variables can hold integers of unlimited magnitude.
Then will print the entire decimal representation of .[note 3]
// Print sqrt(y), without halting
void sqrtForever(unsigned int y)
{
unsigned int result = isqrt(y);
printf("%d.", result); // print result, followed by a decimal point
while (true) // repeat forever ...
{
y = y * 100; // theoretical example: overflow is ignored
result = isqrt(y);
printf("%d", result % 10); // print last digit of result
}
}
The conclusion is that algorithms which compute isqrt()
are computationally equivalent to algorithms which compute sqrt()
.[1]
Basic algorithms
[edit]The integer square root of a non-negative integer can be defined as
For example, because .
Algorithm using linear search
[edit]The following C programs are straightforward implementations.
// Integer square root
// (using linear search, ascending)
unsigned int isqrt(unsigned int y)
{
// initial underestimate, L <= isqrt(y)
unsigned int L = 0;
while ((L + 1) * (L + 1) <= y)
L = L + 1;
return L;
}
// Integer square root
// (using linear search, descending)
unsigned int isqrt(unsigned int y)
{
// initial overestimate, isqrt(y) <= R
unsigned int R = y;
while (R * R > y)
R = R - 1;
return R;
}
Linear search using addition
[edit]In the program above (linear search, ascending) one can replace multiplication by addition, using the equivalence
// Integer square root
// (linear search, ascending) using addition
unsigned int isqrt(unsigned int y)
{
unsigned int L = 0;
unsigned int a = 1;
unsigned int d = 3;
while (a <= y)
{
a = a + d; // (a + 1) ^ 2
d = d + 2;
L = L + 1;
}
return L;
}
Algorithm using binary search
[edit]Linear search sequentially checks every value until it hits the smallest where .
A speed-up is achieved by using binary search instead. The following C-program is an implementation.
// Integer square root (using binary search)
unsigned int isqrt(unsigned int y)
{
unsigned int L = 0;
unsigned int M;
unsigned int R = y + 1;
while (L != R - 1)
{
M = (L + R) / 2;
if (M * M <= y)
L = M;
else
R = M;
}
return L;
}
Numerical example
For example, if one computes using binary search, one obtains the sequence
This computation takes 21 iteration steps, whereas linear search (ascending, starting from ) needs 1414 steps.
Algorithm using Newton's method
[edit]One way of calculating and is to use Heron's method, which is a special case of Newton's method, to find a solution for the equation , giving the iterative formula
The sequence converges quadratically to as .
Stopping criterion
[edit]One can prove[citation needed] that is the largest possible number for which the stopping criterion ensures in the algorithm above.
In implementations which use number formats that cannot represent all rational numbers exactly (for example, floating point), a stopping constant less than 1 should be used to protect against round-off errors.
Domain of computation
[edit]Although is irrational for many , the sequence contains only rational terms when is rational. Thus, with this method it is unnecessary to exit the field of rational numbers in order to calculate , a fact which has some theoretical advantages.
Using only integer division
[edit]For computing for very large integers n, one can use the quotient of Euclidean division for both of the division operations. This has the advantage of only using integers for each intermediate value, thus making the use of floating point representations of large numbers unnecessary. It is equivalent to using the iterative formula
By using the fact that
one can show that this will reach within a finite number of iterations.
In the original version, one has for , and for . So in the integer version, one has and until the final solution is reached. For the final solution , one has and , so the stopping criterion is .
However, is not necessarily a fixed point of the above iterative formula. Indeed, it can be shown that is a fixed point if and only if is not a perfect square. If is a perfect square, the sequence ends up in a period-two cycle between and instead of converging.
Example implementation in C
[edit]// Square root of integer
unsigned int int_sqrt(unsigned int s)
{
// Zero yields zero
// One yields one
if (s <= 1)
return s;
// Initial estimate (must be too high)
unsigned int x0 = s / 2;
// Update
unsigned int x1 = (x0 + s / x0) / 2;
while (x1 < x0) // Bound check
{
x0 = x1;
x1 = (x0 + s / x0) / 2;
}
return x0;
}
Numerical example
[edit]For example, if one computes the integer square root of 2000000 using the algorithm above, one obtains the sequence In total 13 iteration steps are needed. Although Heron's method converges quadratically close to the solution, less than one bit precision per iteration is gained at the beginning. This means that the choice of the initial estimate is critical for the performance of the algorithm.
When a fast computation for the integer part of the binary logarithm or for the bit-length is available (like e.g. std::bit_width
in C++20), one should better start at
which is the least power of two bigger than . In the example of the integer square root of 2000000, , , and the resulting sequence is
In this case only four iteration steps are needed.
Digit-by-digit algorithm
[edit]The traditional pen-and-paper algorithm for computing the square root is based on working from higher digit places to lower, and as each new digit pick the largest that will still yield a square . If stopping after the one's place, the result computed will be the integer square root.
Using bitwise operations
[edit]If working in base 2, the choice of digit is simplified to that between 0 (the "small candidate") and 1 (the "large candidate"), and digit manipulations can be expressed in terms of binary shift operations. With *
being multiplication, <<
being left shift, and >>
being logical right shift, a recursive algorithm to find the integer square root of any natural number is:
def integer_sqrt(n: int) -> int:
assert n >= 0, "sqrt works for only non-negative inputs"
if n < 2:
return n
# Recursive call:
small_cand = integer_sqrt(n >> 2) << 1
large_cand = small_cand + 1
if large_cand * large_cand > n:
return small_cand
else:
return large_cand
# equivalently:
def integer_sqrt_iter(n: int) -> int:
assert n >= 0, "sqrt works for only non-negative inputs"
if n < 2:
return n
# Find the shift amount. See also [[find first set]],
# shift = ceil(log2(n) * 0.5) * 2 = ceil(ffs(n) * 0.5) * 2
shift = 2
while (n >> shift) != 0:
shift += 2
# Unroll the bit-setting loop.
result = 0
while shift >= 0:
result = result << 1
large_cand = (
result + 1
) # Same as result ^ 1 (xor), because the last bit is always 0.
if large_cand * large_cand <= n >> shift:
result = large_cand
shift -= 2
return result
Traditional pen-and-paper presentations of the digit-by-digit algorithm include various optimizations not present in the code above, in particular the trick of pre-subtracting the square of the previous digits which makes a general multiplication step unnecessary. See Methods of computing square roots § Binary numeral system (base 2) for an example.[2]
Karatsuba square root algorithm
[edit]The Karatsuba square root algorithm[3] is a fast algorithm for smaller big-integers if Burnikel-Ziegler Karatsuba division and Karatsuba multiplication are used. Here is an example implementation for 64-bit unsigned integers that uses an unshown 32-bit implementation (u32_isqrt_with_remainder) and takes an input () and produces the integer square root () and the remainder (), where :
/// Performs a Karatsuba square root on a `u64`.
fn u64_isqrt_with_remainder(mut n: u64) -> (u64, u64) {
const HALF_BITS: u32 = u64::BITS >> 1;
const QUARTER_BITS: u32 = u64::BITS >> 2;
const LOWER_HALF_1_BITS: u64 = (1 << HALF_BITS) - 1;
if n <= u32::MAX as u64 {
// If `n` fits in a `u32`, let the `u32` function handle it.
let (s, r) = u32_isqrt_with_remainder(n as u32);
return (s as u64, r as u64);
} else {
// The normalization shift satisfies the Karatsuba square root
// algorithm precondition "a₃ ≥ b/4" where a₃ is the most
// significant quarter of `n`'s bits and b is the number of
// values that can be represented by that quarter of the bits.
//
// b/4 would then be all 0s except the second most significant
// bit (010...0) in binary. Since a₃ must be at least b/4, a₃'s
// most significant bit or its neighbor must be a 1. Since a₃'s
// most significant bits are `n`'s most significant bits, the
// same applies to `n`.
//
// The reason to shift by an even number of bits is because an
// even number of bits produces the square root shifted to the
// left by half of the normalization shift:
//
// sqrt(n << (2 * p))
// sqrt(2.pow(2 * p) * n)
// sqrt(2.pow(2 * p)) * sqrt(n)
// 2.pow(p) * sqrt(n)
// sqrt(n) << p
//
// Shifting by an odd number of bits leaves an ugly sqrt(2)
// multiplied in.
const EVEN_MAKING_BITMASK: u32 = !1;
let normalization_shift = n.leading_zeros() & EVEN_MAKING_BITMASK;
n <<= normalization_shift;
let hi = (n >> HALF_BITS) as u32;
let lo = n & LOWER_HALF_1_BITS;
let (s_prime, r_prime) = u32_isqrt_with_remainder(hi);
let numerator = ((r_prime as u64) << QUARTER_BITS) | (lo >> QUARTER_BITS);
let denominator = (s_prime as u64) << 1;
let q = numerator / denominator;
let u = numerator % denominator;
let mut s = (s_prime << QUARTER_BITS) as u64 + q;
let (mut r, overflow) =
((u << QUARTER_BITS) | (lo & ((1 << QUARTER_BITS) - 1))).overflowing_sub(q * q);
if overflow {
r = r.wrapping_add(2 * s - 1);
s -= 1;
}
let denormalization_shift = normalization_shift / 2;
return (s >> denormalization_shift, r >> denormalization_shift);
}
}
In programming languages
[edit]Some programming languages dedicate an explicit operation to the integer square root calculation in addition to the general case or can be extended by libraries to this end.
- Common Lisp[4]:
(isqrt n)
- Crystal[5]:
Math.isqrt(n)
- Julia[6]:
isqrt(n)
- Maple[7]:
isqrt(n)
- Python[8]:
math.isqrt(n)
- Ruby[9]:
Integer.sqrt(n)
- SageMath[10]:
isqrt(n)
- Scheme[11]:
(exact-integer-sqrt n)
- Tcl[12]:
isqrt($n)
See also
[edit]Notes
[edit]- ^ The square roots of the perfect squares (e.g., 0, 1, 4, 9, 16) are integers. In all other cases, the square roots of positive integers are irrational numbers.
- ^ It is no surprise that the repeated multiplication by 100 is a feature in Jarvis (2006)
- ^ The fractional part of square roots of perfect squares is rendered as 000....
References
[edit]- ^ see 'Methods of computing square roots'.
- ^ Woo, C (June 1985). "Square root by abacus algorithm (archived)". Archived from the original on 2012-03-06.
- ^ Zimmermann, Paul (1999). "Karatsuba Square Root" (PDF). Research report #3805. Inria (published 2006-05-24). Archived from the original (PDF) on 2023-05-11.
- ^ "CLHS: Function SQRT, ISQRT". Common Lisp HyperSpec (TM).
- ^ "Math - Crystal 1.13.2". The Crystal Programming Language API docs (since version 1.2).
{{cite web}}
: CS1 maint: postscript (link) - ^ "Mathematics - The Julia Language". Julia Documentation - The Julia Language (since version 0.3).
{{cite web}}
: CS1 maint: postscript (link) - ^ "iroot- Maple Help". Help - Maplesoft.
- ^ "Mathematical functions". Python Standard Library documentation (since version 3.8).
{{cite web}}
: CS1 maint: postscript (link) - ^ "class Integer - RDoc Documentation". RDoc Documentation (since version 2.5.0).
{{cite web}}
: CS1 maint: postscript (link) - ^ "Elements of the ring ℤ of integers - Standard Commutative Rings". SageMath Documentation.
- ^ "Revised7 Report on the Algorithmic Language Scheme". Scheme Standards (since version R6RS).
{{cite web}}
: CS1 maint: postscript (link) - ^ "mathfunc manual page - Tcl Mathematical Functions". Tcl/Tk 8.6 Manual (since version 8.5).
{{cite web}}
: CS1 maint: postscript (link)
External links
[edit]- Jarvis, Ashley Frazer (2006). "Square roots by subtraction" (PDF). Mathematical Spectrum. 37: 119–122.
- Minsky, Marvin (1967). "9. The Computable Real Numbers". Computation: Finite and Infinite Machines. Prentice-Hall. ISBN 0-13-165563-9. OCLC 0131655639.
- "A geometric view of the square root algorithm".