結果

問題 No.187 中華風 (Hard)
ユーザー Takahiro INOUETakahiro INOUE
提出日時 2019-06-05 17:14:58
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 95 ms / 3,000 ms
コード長 1,341 bytes
コンパイル時間 118 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,264 KB
最終ジャッジ日時 2024-09-22 03:15:27
合計ジャッジ時間 2,786 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
10,880 KB
testcase_01 AC 36 ms
10,880 KB
testcase_02 AC 58 ms
11,008 KB
testcase_03 AC 57 ms
11,136 KB
testcase_04 AC 89 ms
11,136 KB
testcase_05 AC 88 ms
11,264 KB
testcase_06 AC 91 ms
11,264 KB
testcase_07 AC 93 ms
11,264 KB
testcase_08 AC 92 ms
11,136 KB
testcase_09 AC 94 ms
11,264 KB
testcase_10 AC 95 ms
11,136 KB
testcase_11 AC 89 ms
11,136 KB
testcase_12 AC 88 ms
11,264 KB
testcase_13 AC 33 ms
10,880 KB
testcase_14 AC 33 ms
11,008 KB
testcase_15 AC 67 ms
11,008 KB
testcase_16 AC 69 ms
11,136 KB
testcase_17 AC 30 ms
10,880 KB
testcase_18 AC 33 ms
10,880 KB
testcase_19 AC 30 ms
10,880 KB
testcase_20 AC 76 ms
11,136 KB
testcase_21 AC 30 ms
10,752 KB
testcase_22 AC 91 ms
11,264 KB
testcase_23 AC 30 ms
10,752 KB
testcase_24 AC 30 ms
10,880 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 fraenkel(A, M):
    '''
        solve
            x \equiv a_1 (mod m_1), ... x \equiv a_k (mod m_k)
        by applying the Fraenkel's Algorithm
        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):
        s, _, g = extended_euclid(mod, m)
        q, r = divmod(a - x, g)
        if r != 0:
            return (0, -1)
        temp = m // g
        u = (q * s) % (temp)
        x += mod * u
        mod *= temp
        x %= mod
    return (x, mod)

N = int(input())
MOD = 10**9 + 7
A = []
M = []
for _ in range(N):
    a, m = map(int, input().split())
    A.append(a)
    M.append(m)
x, m = fraenkel(A, M)
if (x, m) == (0, -1):
    x = -1
elif x == 0 and m != -1:
    x = reduce(lcm, M)
print(x % MOD if x != -1 else x)
0