結果
問題 | 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
|
実行時間 | 637 ms / 2,000 ms |
コード長 | 767 bytes |
コンパイル時間 | 436 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 88,320 KB |
最終ジャッジ日時 | 2024-04-27 21:16:07 |
合計ジャッジ時間 | 15,359 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 524 ms
44,232 KB |
testcase_01 | AC | 528 ms
44,364 KB |
testcase_02 | AC | 517 ms
44,624 KB |
testcase_03 | AC | 511 ms
44,108 KB |
testcase_04 | AC | 512 ms
44,236 KB |
testcase_05 | AC | 527 ms
44,496 KB |
testcase_06 | AC | 530 ms
44,368 KB |
testcase_07 | AC | 515 ms
44,616 KB |
testcase_08 | AC | 513 ms
44,236 KB |
testcase_09 | AC | 515 ms
44,364 KB |
testcase_10 | AC | 596 ms
45,136 KB |
testcase_11 | AC | 526 ms
44,492 KB |
testcase_12 | AC | 637 ms
88,320 KB |
testcase_13 | AC | 533 ms
44,240 KB |
testcase_14 | AC | 545 ms
45,388 KB |
testcase_15 | AC | 536 ms
44,236 KB |
testcase_16 | AC | 562 ms
45,132 KB |
testcase_17 | AC | 531 ms
44,108 KB |
testcase_18 | AC | 618 ms
88,120 KB |
testcase_19 | AC | 545 ms
51,824 KB |
testcase_20 | AC | 569 ms
47,876 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)