結果

問題 No.186 中華風 (Easy)
ユーザー Takahiro INOUETakahiro INOUE
提出日時 2019-06-04 10:47:02
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 31 ms / 2,000 ms
コード長 1,259 bytes
コンパイル時間 145 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-07-19 18:34:48
合計ジャッジ時間 1,380 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 27 ms
10,752 KB
testcase_01 AC 27 ms
10,624 KB
testcase_02 AC 31 ms
10,752 KB
testcase_03 AC 28 ms
10,624 KB
testcase_04 AC 28 ms
10,752 KB
testcase_05 AC 25 ms
10,624 KB
testcase_06 AC 25 ms
10,624 KB
testcase_07 AC 26 ms
10,752 KB
testcase_08 AC 29 ms
10,752 KB
testcase_09 AC 29 ms
10,752 KB
testcase_10 AC 27 ms
10,752 KB
testcase_11 AC 25 ms
10,624 KB
testcase_12 AC 26 ms
10,752 KB
testcase_13 AC 26 ms
10,752 KB
testcase_14 AC 28 ms
10,752 KB
testcase_15 AC 25 ms
10,624 KB
testcase_16 AC 27 ms
10,624 KB
testcase_17 AC 26 ms
10,624 KB
testcase_18 AC 27 ms
10,752 KB
testcase_19 AC 27 ms
10,752 KB
testcase_20 AC 28 ms
10,624 KB
testcase_21 AC 27 ms
10,624 KB
testcase_22 AC 26 ms
10,752 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.py:20: SyntaxWarning: invalid escape sequence '\e'
  '''

ソースコード

diff # 

from functools import reduce

def gcd(a, b):
    while b:
        a, b = b, a%b
    return a

def lcm(a, b):
	return a * b // gcd (a, b)

def extended_euclid(a, b):
    x1, y1, m = 1, 0, a
    x2, y2, n = 0, 1, b
    while m % n != 0:
        q, r = divmod(m, n)
        x1, y1, m, x2, y2, n = x2, y2, n, x1 - q * x2, y1 - q * y2, r
    return (x2, y2, n)

def chinese_reminder(A, M):
    '''
        solve
            x \equiv a_1 (mod m_1), ... x \equiv a_k (mod m_k)
        by applying the Chinese reminder theorem
        Input: 
            A = [a_1, ..., a_k]: a list
            M = [m_1, ..., m_k]: a list.
        Output:
            Returns the tuple (x, mod) of the solution x and the modulus mod = m_1 ... m_k if exists
            else (0, -1).
    '''
    # initialize
    x = 0
    mod = 1
    for a, m in zip(A, M):
        u, v, g = extended_euclid(mod, m)
        q, r = divmod(a - x, g)
        if r != 0:
            return (0, -1)
        x += q * mod * u
        mod *= m // g
        x %= mod
    return (x, mod)

A = []
M = []
for _ in range(3):
    a, m = map(int, input().split())
    A.append(a)
    M.append(m)
x, m = chinese_reminder(A, M)
if (x, m) == (0, -1):
    x = -1
elif x == 0 and m != -1:
    x = reduce(lcm, M)
print(x)
0