結果
問題 | No.990 N×Mマス計算(Kの倍数) |
ユーザー | maspy |
提出日時 | 2020-02-19 16:14:10 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 601 ms / 2,000 ms |
コード長 | 767 bytes |
コンパイル時間 | 554 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 88,108 KB |
最終ジャッジ日時 | 2024-11-16 03:00:39 |
合計ジャッジ時間 | 15,367 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 530 ms
44,364 KB |
testcase_01 | AC | 492 ms
44,108 KB |
testcase_02 | AC | 479 ms
44,104 KB |
testcase_03 | AC | 497 ms
44,116 KB |
testcase_04 | AC | 501 ms
44,236 KB |
testcase_05 | AC | 498 ms
44,240 KB |
testcase_06 | AC | 498 ms
44,488 KB |
testcase_07 | AC | 497 ms
44,496 KB |
testcase_08 | AC | 491 ms
44,488 KB |
testcase_09 | AC | 493 ms
44,232 KB |
testcase_10 | AC | 557 ms
44,880 KB |
testcase_11 | AC | 507 ms
44,496 KB |
testcase_12 | AC | 581 ms
88,108 KB |
testcase_13 | AC | 498 ms
44,620 KB |
testcase_14 | AC | 523 ms
45,896 KB |
testcase_15 | AC | 541 ms
44,108 KB |
testcase_16 | AC | 520 ms
45,132 KB |
testcase_17 | AC | 518 ms
44,368 KB |
testcase_18 | AC | 601 ms
87,588 KB |
testcase_19 | AC | 525 ms
52,144 KB |
testcase_20 | AC | 556 ms
48,260 KB |
ソースコード
#!/usr/bin/env python3 # %% stdin = open(0) # %% import numpy as np # %% N, M, K = map(int, stdin.readline().split()) op = stdin.read(1) BA = np.fromstring(stdin.read(), np.int64, sep=' ') # %% B = BA[:M] A = BA[M:] # %% def solve_add(A, B): A %= K A.sort() target = (-B) % K cnt = np.searchsorted(A, target, 'right') - np.searchsorted(A, target, 'left') return cnt.sum() # %% def solve_mult(A, B): B_keys, B_nums = np.unique(np.gcd(B, K), return_counts=True) A_keys, A_nums = np.unique(np.gcd(A, K), return_counts=True) keys = np.multiply.outer(A_keys, B_keys) nums = np.multiply.outer(A_nums, B_nums) return ((keys % K == 0) * nums).sum() # %% answer = solve_add(A, B) if op == '+' else solve_mult(A, B) print(answer)