結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー nehan_der_thalnehan_der_thal
提出日時 2020-02-27 21:15:16
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,475 bytes
コンパイル時間 375 ms
コンパイル使用メモリ 82,360 KB
実行使用メモリ 242,748 KB
最終ジャッジ日時 2024-11-16 03:28:42
合計ジャッジ時間 17,234 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
61,948 KB
testcase_01 AC 41 ms
171,248 KB
testcase_02 AC 54 ms
69,640 KB
testcase_03 AC 39 ms
242,748 KB
testcase_04 AC 44 ms
68,692 KB
testcase_05 AC 38 ms
182,092 KB
testcase_06 AC 51 ms
71,036 KB
testcase_07 AC 49 ms
63,168 KB
testcase_08 AC 39 ms
54,992 KB
testcase_09 AC 38 ms
53,872 KB
testcase_10 AC 101 ms
89,800 KB
testcase_11 TLE -
testcase_12 TLE -
testcase_13 TLE -
testcase_14 WA -
testcase_15 WA -
testcase_16 AC 149 ms
89,044 KB
testcase_17 WA -
testcase_18 TLE -
testcase_19 TLE -
testcase_20 AC 198 ms
227,396 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
def prime_decomposition(n):
    i = 2
    table = defaultdict(int)
    while i * i <= n:
        c = 0
        while n % i == 0:
            n /= i
            c+=1
        if c!=0:
            table[i] = c
        i += 1
    if n > 1:
        table[int(n)] = 1
    return table
N, M, K = map(int, input().split())
R = 0
table = prime_decomposition(K)
bs = []
fs = []
bbs = dict()
ffs = dict()
for i in range(N+1):
    if i == 0:
        X = input().split()
        e = X[0] == '+'
        if e:
            d = defaultdict(int)
            for x in X[1:]:
                d[K-int(x)%K] += 1
        else:
            bs = list(map(lambda x: prime_decomposition(int(x)), X[1:]))
            d = defaultdict(int)
            for b in bs:
                d[1] += 1
                yakusus = [1]
                for k in b:
                    yn = len(yakusus)
                    for j in range(yn):
                        for l in range(b[k]):
                            yakusus.append(yakusus[j]*(k**(l+1)))
                            d[yakusus[j]*(k**(l+1))] += 1
        continue
    x = int(input())
    if e:
        R += d[x%K]
    else:
        fs.append(prime_decomposition(x))
if e:
    print(R)
else:
    R = 0
    for b in fs:
        r = K
        for key in b:
            if key in table:
                r //= key**min(table[key], b[key])
#        print(r, K)
        R += d[r]
    #        ffs[key].sort()

    print(R)
0