-
-
Notifications
You must be signed in to change notification settings - Fork 1.8k
GH-1005: Pell's equation #1388
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
GH-1005: Pell's equation #1388
Changes from all commits
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -8,5 +8,6 @@ converted.pdf | |
/public | ||
.firebase/ | ||
firebase-debug.log | ||
.idea/ | ||
|
||
authors.json | ||
authors.json |
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,201 @@ | ||
--- | ||
tags: | ||
- Original | ||
--- | ||
|
||
# Pell's Equation (Pell-Fermat Equation) | ||
|
||
## Statement | ||
We are given a natural number $d$. We need to find the smallest positive integer $x$ such that $x^{2} - d \cdot y^{2} = 1$ for some positive integer $y$. | ||
|
||
Alternative formulation: We want to find all the possible solutions of the equation $x^{2} - d \cdot y^{2} = 1$. | ||
|
||
## Solution | ||
Here we will consider the case when $d$ is not a perfect square and $d>1$. The case when $d$ is a perfect square is trivial. | ||
We can even assume that $d$ is square-free (i.e. it is not divisible by the square of any prime number) as we can absorb the factors of $d$ into $y$. | ||
|
||
$x^{2} - d \cdot y^{2} = ( x + \sqrt{d} \cdot y ) ( x - \sqrt{d} \cdot y ) = 1$ | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. This equation looks like it just comes out of nowhere. As in, it isn't connected with the paragraphs that are immediately adjusted to it. I would add some explanation on what are we doing here exactly. Also consider wrapping this in |
||
|
||
The first part $( x + \sqrt{d} \cdot y )$ is always greater than 1. And the second part $( x - \sqrt{d} \cdot y )$ is always less than 1, since the product is 1. | ||
|
||
We will prove that all solutions to Pell's equation are given by powers of the smallest positive solution. Let's assume it to be | ||
$x_{0} + y_{0} \cdot \sqrt{d}$ | ||
|
||
We use method of descent to prove it. | ||
Suppose there is a solution $u + v \cdot \sqrt{d}$ such that $u^{2} - d \cdot v^{2} = 1$ and is not a power of $( x_{0} + \sqrt{d} \cdot y_{0} )$ | ||
Then it must lie between two powers of $( x_{0} + \sqrt{d} \cdot y_{0} )$. | ||
i.e, For some n, $( x_{0} + \sqrt{d} \cdot y_{0} )^{n} < u + v \cdot \sqrt{d} < ( x_{0} + \sqrt{d} \cdot y_{0} )^{n+1}$ | ||
|
||
Multiplying the above inequality by $( x_{0} - \sqrt{d} \cdot y_{0} )^{n}$,(which is > 0 and < 1) we get | ||
|
||
$1 < (u + v \cdot \sqrt{d})( x_{0} - \sqrt{d} \cdot y_{0} )^{n} < ( x_{0} + \sqrt{d} \cdot y_{0} )$ | ||
Because both $(u + v \cdot \sqrt{d})$ and $( x_{0} - \sqrt{d} \cdot y_{0} )^{n}$ have norm 1, their product is also a solution. | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. You use the term "norm" here, but it's not defined. We should add a definition + an explanation why we care about it (primarily because it is multiplicative). |
||
But this contradicts our assumption that $( x_{0} + \sqrt{d} \cdot y_{0} )$ is the smallest solution. Therefore, there is no solution between $( x_{0} + \sqrt{d} \cdot y_{0} )^{n}$ and $( x_{0} + \sqrt{d} \cdot y_{0} )^{n+1}$. | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. We defined the smallest solution by saying that it is the solution with the smallest positive |
||
|
||
Hence, we conclude that all solutions are given by $( x_{0} + \sqrt{d} \cdot y_{0} )^{n}$ for some integer $n$. | ||
|
||
## Finding the smallest positive solution | ||
### Expressing the solution in terms of continued fractions | ||
We can express the solution in terms of continued fractions. The continued fraction of $\sqrt{d}$ is periodic. Let's assume the continued fraction of $\sqrt{d}$ is $[a_{0}; \overline{a_{1}, a_{2}, \ldots, a_{r}}]$. The smallest positive solution is given by the convergent $[a_{0}; a_{1}, a_{2}, \ldots, a_{r}]$ where $r$ is the period of the continued fraction. | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Let's add a link to continued fractions article. Ideally, I would also add explanation on why it is a solution at all, and why is it the smallest possible. There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Also, the paragraph says that "the smallest solution is given by ...", but doesn't tell that, specifically, the solution is |
||
|
||
The convergents $p_{n}/q_{n}$ are the rational approximations to $\sqrt{d}$ obtained by truncating the continued fraction expansion at each stage. These convergents can be computed recursively. | ||
|
||
Check whether the convergent satisfies the Pell's equation. If it does, then the convergent is the smallest positive solution. | ||
|
||
Let's take an example to understand this by solving the equation $x^{2} - 2 \cdot y^{2} = 1$. | ||
$\sqrt{2} = [1; \overline{2}] = 1 + 1/(2 + 1/(2 + 1/(2+ ...)))$. The convergents are $1/1, 3/2, 7/5, 17/12, 41/29, 99/70, \ldots$. | ||
Now check for each convergent whether it satisfies the Pell's equation. The smallest positive solution is $3/2$. | ||
|
||
### How to calculate the continued fraction of $\sqrt{d}$? | ||
Let's find the continued fraction of $\sqrt{7}$. | ||
|
||
$\sqrt{7} \approx 2.6457 = 2 + 0.6457$ | ||
|
||
$a_{0} = 2$ | ||
|
||
Subtract $a_{0}$ from the number and take the reciprocal of the remaining number. | ||
|
||
That is, we calculate ${1\over \sqrt{7} - 2} \approx 1.5486$. The integer part $a_{1}$ is $1$. | ||
So: | ||
$\sqrt{7}=2+\cfrac{1}{1+\cfrac1{\vdots}}$ | ||
Where we haven't calculated the $( \vdots )$ part yet. | ||
To get that, we subtract $a_{1}$ from the number and take the reciprocal of the remaining number. That is, we calculate ${1\over 1.5486 - 1} \approx 1.8228$. The integer part $a_{2}$ is $1$. | ||
So: | ||
$\sqrt{7}=2+\cfrac{1}{1+\cfrac1{1+\cfrac1{\vdots}}}$ | ||
Now ${1\over 1.8228 - 1} \approx 1.2153$. So $a_{3} = 1$. | ||
Continuing this process, ${1\over 1.2153 - 1} \approx 4.645$. So $a_{4} = 4$. | ||
|
||
$$\sqrt{7}=2+\cfrac{1}{1+\cfrac1{1+\cfrac1{1+\cfrac4{\vdots}}}}$$ | ||
|
||
we get the continued fraction of $\sqrt{7}$ as $[2; 1, 1, 4, 1, 1, 4, \ldots]$. | ||
|
||
This can also be calculated using [integer based calculation(continued fractions)](https://cp-algorithms.com/algebra/continued-fractions.html) | ||
|
||
### Finding the solution using Chakravala method | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. This looks complicated. Are there specific reasons to do it over continued fractions variant? I find the exposition very hard to read (largely because of misrenders, though). Unless it's in some way better than continued fractions method, I'd consider dropping the section altogether. |
||
The Chakravala method is an ancient Indian algorithm to solve Pell's equation. It is based on the Brahmagupta's identity of quadratic decomposition | ||
$(x_{1}^{2} - n \cdot y_{1}^{2}) \cdot (x_{2}^{2} - n \cdot y_{2}^{2}) = (x_{1} \cdot x_{2} + n \cdot y_{1} \cdot y_{2})^{2} - n \cdot (x_{1} \cdot y_{2} + x_{2} \cdot y_{1})^{2}$ | ||
$(x_{1}^{2} - n \cdot y_{1}^{2}) \cdot (x_{2}^{2} - n \cdot y_{2}^{2}) = (x_{1} \cdot x_{2} - n \cdot y_{1} \cdot y_{2})^{2} - n \cdot (x_{1} \cdot y_{2} - x_{2} \cdot y_{1})^{2}$ | ||
|
||
And Bhaskara's Lemma: | ||
Bhaskar1312 marked this conversation as resolved.
Show resolved
Hide resolved
|
||
If $x^{2} - n \cdot y^{2} = k$, then $( \frac{ m \cdot x + n \cdot y }{k})^{2} - n \cdot ( \frac{ x + m \cdot y }{k})^{2} = \frac{m^2 - n}{k}$ | ||
|
||
Using above Brahmagupta's identity, If $(x_{1}, y_{1}, k_{1})$ and $(x_{2}, y_{2}, k_{2})$ satisfy $(x_{1}^{2} - y_1^{2}) \cdot (x_{2}^{2} - y_2^{2}) = k_{1} \cdot k_{2}$, then $(x_{1} \cdot x_{2} + n \cdot y_{1} \cdot y_{2}, x_{1} \cdot y_{2} + y_{1} \cdot x_{2}, k_{1} \cdot k_{2})$ is also a solution of $(x_{1} \cdot x_{2} + n \cdot y_{1} \cdot y_{2})^{2} - n \cdot (x_{1} \cdot y_{2} + x_{2} \cdot y_{1})^{2} = k_{1} \cdot k_{2}$ | ||
|
||
#### Steps | ||
1. Initialization:Choose an initial solution $(p_{0}, q_{0}, m_{0})$ where $p_{0}$ and $q_{0}$ are co-prime such that $p_{0}^{2} - N \cdot q_{0}^{2} = m_{0}$. Typically, start with $p_{0} = \lfloor \sqrt N \rfloor$, $q_{0} = 1$, $m_{0} = p_0^2 - N$. | ||
2. Key step: Find $x_{1}$ such that: $q_{0} \cdot x_{1} \equiv -p_{0} \pmod {\lvert m_{0}\rvert}$ and $\lvert x_{1}^2 - N \rvert$ is minimized. | ||
Update the triple $(p_{1}, q_{1}, m_{1}) = ( \frac{x_{1} \cdot p_{0} + N \cdot q_{0}}{\lvert m_{0} \rvert}, \frac{p_{0} + x_{1} \cdot q_{0}}{\lvert m_{0} \rvert}, \frac{x_1^{2} - N}{m_{0}})$. | ||
3. Termination: When $m_{k}=1$, the values of $p_{k}$ and $q_{k}$ are the smallest positive solution of the Pell's equation. | ||
|
||
##### Example | ||
Let's solve the equation $x^{2} - 13 \cdot y^{2} = 1$ using Chakravala method. | ||
1. Start with $(p_{0}, q_{0}, m_{0}) = (3, 1, -4)$ because $3^2 - 13 \cdot1^2 = -4$. | ||
|
||
2. Find $x_{1}$ such that $x_{1} \equiv -3 \pmod {4}$ and $\lvert x_{1}^2 - 13 \rvert$ is minimized. | ||
We get $x_{1} = 1$. Update the triple $(p_{1}, q_{1}, m_{1}) = ( \frac{1 \cdot 3 + 13 \cdot 1}{4}, \frac{3 + 1 \cdot 1}{4}, \frac{1^{2} - 13}{-4}) = (4, 1, 3)$. | ||
3. Substituting $(p_{1}, q_{1}, k_{1}) = (4, 1, 3)$ in key step, we get $x_{2} \equiv -4 \pmod 3$ and minimize $\lvert x_{2}^2 - 13 \rvert$ i.e, $x_{2} = 2$. Update the triple $(p_{2}, q_{2}, m_{2}) = ( \frac{2 \cdot 4 + 13 \cdot 1}{3}, \frac{4 + 2 \cdot 1}{3}, \frac{2^{2} - 13}{-3}) = (7, 2, -3)$. | ||
4. Substituting $(p_{2}, q_{2}, m_{2}) = (7, 2, -3)$ in key step, we get $2 \cdot x_{3} \equiv -7 \pmod 3$ and minimize $\lvert x_{3}^2 - 13 \rvert$ i.e, $x_{3} = 4$. Update the triple $(p_{3}, q_{3}, m_{3}) = ( \frac{4 \cdot 7 + 13 \cdot 2}{3}, \frac{7 + 4 \cdot 2}{3}, \frac{4^{2} - 13}{-3}) = (18, 5, -1)$. | ||
5. Substituting $(p_{3}, q_{3}, m_{3}) = (18, 5, -1)$ in key step, we get $5 \cdot x_{4} \equiv -18 \pmod 1$ and minimize $\lvert x_{4}^2 - 13 \rvert$ i.e, $x_{4} = 4$. Update the triple $(p_{4}, q_{4}, m_{4}) = ( \frac{4 \cdot 18 + 13 \cdot 5}{1}, \frac{18 + 4 \cdot 5}{1}, \frac{4^{2} - 13}{-1}) = (137, 38, -3)$. | ||
6. Substituting $(p_{4}, q_{4}, m_{4}) = (137, 38, -3)$ in key step, we get $38 \cdot x_{5} \equiv -137 \pmod 3$ and minimize $\lvert x_{5}^2 - 13 \rvert$ i.e, $x_{5} = 2$. Update the triple $(p_{5}, q_{5}, m_{5}) = ( \frac{2 \cdot 137 + 13 \cdot 38}{3}, \frac{137 + 2 \cdot 38}{3}, \frac{2^{2} - 13}{-3}) = (256, 71, 3)$. | ||
7. Substituting $(p_{5}, q_{5}, m_{5}) = (256, 71, 3)$ in key step, we get $71 \cdot x_{6} \equiv -256 \pmod 3$ and minimize $\lvert x_{6}^2 - 13 \rvert$ i.e, $x_{6} = 4$. Update the triple $(p_{6}, q_{6}, m_{6}) = ( \frac{4 \cdot 256 + 13 \cdot 71}{3}, \frac{256 + 4 \cdot 71}{3}, \frac{4^{2} - 13}{3}) = (649, 180, 1)$. | ||
|
||
## Implementation | ||
```cpp | ||
bool isSquare(long long n) { | ||
long long sqrtN = (long long)sqrt(n); | ||
return sqrtN * sqrtN == n; | ||
} | ||
|
||
long long mod(long long a, long long b) { | ||
return (a % b + b) % b; | ||
} | ||
|
||
long long modInv(long long a, long long b) { | ||
long long b0 = b, x0 = 0, x1 = 1; | ||
if (b == 1) return 1; | ||
while (a > 1) { | ||
long long q = a / b; | ||
long long temp = b; | ||
b = a % b; | ||
a = temp; | ||
temp = x0; | ||
x0 = x1 - q * x0; | ||
x1 = temp; | ||
} | ||
if (x1 < 0) x1 += b0; | ||
return x1; | ||
} | ||
|
||
|
||
// Chakravala method for solving Pell's equation | ||
pair<long long, long long> chakravala(int n) { | ||
// Check if n is a perfect square | ||
if (isSquare(n)) { | ||
throw invalid_argument("n is a perfect square. No solutions exist for Pell's equation."); | ||
} | ||
|
||
// Initial values | ||
double sqrt_n = sqrt(n); | ||
long long a = (long long)floor(sqrt_n); // Initial a | ||
long long b = 1; // Initial b | ||
long long k = a * a - n; // Initial k | ||
|
||
int steps = 0; // Step counter for iterations | ||
|
||
// Repeat until k = 1 | ||
while (k != 1) { | ||
long long absK = abs(k); | ||
|
||
// Find m such that k | (a + bm), and minimize |m^2 - n| | ||
long long m; | ||
if (absK == 1) { | ||
m = (long long)round(sqrt(n)); // round to nearest integer | ||
} else { | ||
long long r = mod(-a, absK); // Negative of a mod(k) // (a + m*b)/|k| | ||
long long s = modInv(b, absK); // Modular inverse of b mod(k) | ||
m = mod(r * s, absK); // Compute m for (a + b*m) mod(k) = 0 | ||
|
||
// Approximate value of m | ||
// m = m + ((long long)floor((sqrt_n - m) / absK)) * absK; | ||
|
||
// Adjust m to ensure m < sqrt(n) < m + k | ||
while (m > sqrt(n)) m -= absK; | ||
while (m + absK < sqrt_n) m += absK; | ||
|
||
// Select closest value to n | ||
if (abs(m * m - n) > abs((m + absK) * (m + absK) - n)) { | ||
m = m + absK; | ||
} | ||
} | ||
|
||
// Print the current triple | ||
cout << "[a = " << a << ", b = " << b << ", k = " << k << "]" << endl; | ||
|
||
// Update a, b, k using the recurrence relations | ||
long long alpha = a; | ||
a = (m * a + n * b) / absK; | ||
b = (alpha + m * b) / absK; | ||
k = (m * m - n) / k; | ||
|
||
// Increment step counter | ||
steps++; | ||
} | ||
|
||
// Print final result | ||
cout << a << "^2 - " << n << " x " << b << "^2 = 1 in " << steps << " calculations." << endl; | ||
|
||
// Return the solution as a pair (a, b) | ||
return {a, b}; | ||
} | ||
|
||
``` | ||
|
||
## References | ||
- [Pell's equation - Wikipedia](https://en.wikipedia.org/wiki/Pell%27s_equation) | ||
- [Periodic Continued Fractions](https://en.wikipedia.org/wiki/Periodic_continued_fraction) | ||
- [Chakravala Method](http://publications.azimpremjifoundation.org/1630/1/3_The%20Chakravala%20Method.pdf) | ||
- [Pythagorean triples and Pell's equations - Codeforces](https://codeforces.com/blog/entry/116313) | ||
|
||
## Problems | ||
- [Project Euler 66](https://projecteuler.net/problem=66) | ||
- [Hackerrank ProjectEuler-066](https://www.hackerrank.com/contests/projecteuler/challenges/euler066/problem) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Do not include this
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
.gitgnore just ignores files to not include them when we do
git add .
So, that won't affect anything
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think we should remove this, unless it is something that is really needed for the project at large, rather than specific developer.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why is it here again 🤔