Skip to content

Commit

Permalink
added diophantine
Browse files Browse the repository at this point in the history
  • Loading branch information
rohitsh16 committed Oct 2, 2019
1 parent 469484e commit aa83aca
Show file tree
Hide file tree
Showing 2 changed files with 82 additions and 0 deletions.
82 changes: 82 additions & 0 deletions diophantine/diophantine.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,82 @@
#include <bits/stdc++.h>

// find all solution ax + by = c for (x, y) in range of [minx : maxx] and [miny : maxy]

using namespace std;

int x, y, g, lx, rx;

int gcd (int a, int b, int & x, int & y) {
if (a == 0) {
x = 0; y = 1;
return b;
}
int x1, y1;
int d = gcd (b%a, a, x1, y1);
x = y1 - (b / a) * x1;
y = x1;
return d;
}

bool find_any_solution (int a, int b, int c, int & x0, int & y0, int & g) {
g = gcd (abs(a), abs(b), x0, y0);
if (c % g != 0)
return false;
x0 *= c / g;
y0 *= c / g;
if (a < 0) x0 *= -1;
if (b < 0) y0 *= -1;
return true;
}

void shift_solution(int &x,int &y,int a,int b,int cnt){
x += cnt * b;
y -= cnt * a;
}

//It returns the number of solutions that exists.

int find_all_solutions(int a, int b, int c, int minx, int maxx, int miny, int maxy) {
if (!find_any_solution(a, b, c, x, y, g))
return 0;
a /= g;
b /= g;

int sign_a = a>0 ? +1 : -1;
int sign_b = b>0 ? +1 : -1;

shift_solution(x, y, a, b, (minx - x) / b);
if (x < minx) shift_solution(x, y, a, b, sign_b);
if (x > maxx) return 0;
int lx1 = x;

shift_solution(x, y, a, b, (maxx - x) / b);
if (x > maxx) shift_solution(x, y, a, b, -sign_b);
int rx1 = x;

shift_solution(x, y, a, b, - (miny - y) / a);
if (y < miny) shift_solution(x, y, a, b, -sign_a);
if (y > maxy) return 0;
int lx2 = x;

shift_solution(x, y, a, b, - (maxy - y) / a);
if (y > maxy) shift_solution(x, y, a, b, sign_a);
int rx2 = x;

if (lx2 > rx2)
swap (lx2, rx2);

lx = max (lx1, lx2);
rx = min (rx1, rx2);

return (rx - lx) / abs(b) + 1;
}

int main(){
int a, b, c, maxx, maxy, minx, miny;
cin >> a >> b >> c >> minx >> maxx >> miny >> maxy;
if (find_all_solutions(a,b,c,minx,maxx,miny,maxy)!= 0 ) {
for (int i=lx;i<=rx;i+=b)
cout<<i<<" "<<(c-a*i)/b<<endl;
}
}
Empty file added series_sum/series.txt
Empty file.

0 comments on commit aa83aca

Please sign in to comment.