help with exgcd problem

Блог пользователя om1429888

Автор om1429888, история, 22 месяца назад, По-английски

https://codeforces.com/problemsets/acmsguru/problem/99999/106 can someone please help me solve this?

om1429888
22 месяца назад
6

Комментарии (4)

Показать архивные | Написать комментарий?

islingr

22 месяца назад, # |

← Rev. 5 →

Note that if $$$\gcd(a, b) \not\mid c$$$, no solution exists. Otherwise, divide $$$a, b, c$$$ by this $$$\gcd$$$, now we assume $$$\gcd(a, b) = 1$$$.

Let $$$x_0, y_0$$$ be any integer solution of the equation (which can be found using Euclid's algorithm).

One can prove that all solutions of the equation must be of the form $$$(x_0 - b n, y_0 + a n)$$$ for some integer $$$n$$$.

Now, we end up with the problem of counting integer $$$n$$$ such that $$$x_1 \le x_0 - b n \le x_2$$$, $$$y_1 \le y_0 + a n \le y_2$$$.

This can be done by just getting min and max bounds on $$$n$$$ using the above inequalities. Make sure to take care of all the cases such as $$$a = 0$$$, $$$a < 0$$$, when the upper bound is less than the lower bound, etc..

→ Ответить

om1429888

22 месяца назад, # ^ |

how to find the bounds?

→ Ответить

islingr

22 месяца назад, # ^ |

If $$$b > 0$$$, you have

$$$x_1 \le x_0 - b n \le x_2 \iff x_0 - x_1 \ge b n \ge x_0 - x_2 \iff \frac{x_0 - x_1}{b} \ge n \ge \frac{x_0 - x_2}{b}$$$

If b < 0, the inequalities are reversed.

You similarly get another relation from the other inequality.

Take note of the fact that what I have written above is real division, not integer division.

→ Ответить

YMSeah

22 месяца назад, # |

← Rev. 2 →

You can take reference from CP-algorithms's tutorial on Linear Diophantine Equations. They have the exact solution. There are a lot of edge cases to consider like if a == 0 or b == 0 etc, or a < 0 or b < 0.

Python Solution. Note that '//' is a floor divide in Python


def readIntArr():
    return [int(x) for x in input().split()]

def extended_gcd(a, b):
    """returns gcd(a, b), s, r s.t. a * s + b * r == gcd(a, b)"""
    s, old_s = 0, 1
    r, old_r = b, a
    while r:
        q = old_r // r
        old_r, r = r, old_r - q * r
        old_s, s = s, old_s - q * s
    return old_r, old_s, (old_r - old_s * a) // b if b else 0

def roundup(x):
    if x == int(x):
        return int(x)
    else:
        y = int(abs(x))
        if x < 0:
            return -y
        else:
            return y + 1

def rounddown(x):
    if x == int(x):
        return int(x)
    else:
        y = int(abs(x))
        if x < 0:
            return -y - 1
        else:
            return y

def solve():
    a, b, c = readIntArr()
    x1, x2 = readIntArr()
    y1, y2 = readIntArr()
    
    c *= -1
    
    if a == 0 and b == 0:
        if c == 0:
            ans = (x2 - x1 + 1) * (y2 - y1 + 1)
        else:
            ans = 0
        print(ans)
        return
    if a == 0:
        if c % b == 0 and y1 <= c // b <= y2:
            ans = 1
        else:
            ans = 0
        print(ans)
        return
    if b == 0:
        if c % a == 0 and x1 <= c // a <= x2:
            ans = 1
        else:
            ans = 0
        print(ans)
    
    g, xg, yg = extended_gcd(a, b)
    
    if c % g != 0:
        ans = 0
        print(ans)
        return

    x0 = xg * (c // g)
    y0 = yg * (c // g)
    
    
    k = roundup((x1 - x0) / (b // g))
    xlow = x0 + k * (b // g)
    k = rounddown((x2 - x0) / (b // g))
    xhigh = x0 + k * (b // g)
    
    if xlow > xhigh:
        ans = 0
        print(ans)
        return
    
    ylow = (c - a * xlow) // b
    yhigh = (c - a * xhigh) // b
    if ylow > yhigh:
        ylow, yhigh = yhigh, ylow
    
    k = roundup((y1 - y0) / (a // g))
    ylow2 = y0 + k * (a // g)
    k = rounddown((y2 - y0) / (a // g))
    yhigh2 = y0 + k * (a // g)
    
    ylow3 = max(ylow, ylow2)
    yhigh3 = min(yhigh, yhigh2)
    
    if ylow3 > yhigh3:
        ans = 0
        print(ans)
        return
    
    ans = 1 + (yhigh3 - ylow3) // abs(a // g)
        
    print(ans)
            
solve()

→ Ответить