結果
問題 | 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
|
実行時間 | 242 ms / 2,000 ms |
コード長 | 767 bytes |
コンパイル時間 | 104 ms |
コンパイル使用メモリ | 10,844 KB |
実行使用メモリ | 77,136 KB |
最終ジャッジ日時 | 2023-08-10 03:43:14 |
合計ジャッジ時間 | 7,144 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 132 ms
29,744 KB |
testcase_01 | AC | 137 ms
29,868 KB |
testcase_02 | AC | 133 ms
29,836 KB |
testcase_03 | AC | 133 ms
29,708 KB |
testcase_04 | AC | 133 ms
29,844 KB |
testcase_05 | AC | 134 ms
29,800 KB |
testcase_06 | AC | 134 ms
29,884 KB |
testcase_07 | AC | 136 ms
29,884 KB |
testcase_08 | AC | 134 ms
29,728 KB |
testcase_09 | AC | 134 ms
29,744 KB |
testcase_10 | AC | 201 ms
32,320 KB |
testcase_11 | AC | 151 ms
32,476 KB |
testcase_12 | AC | 242 ms
77,112 KB |
testcase_13 | AC | 149 ms
31,852 KB |
testcase_14 | AC | 174 ms
32,524 KB |
testcase_15 | AC | 152 ms
31,432 KB |
testcase_16 | AC | 156 ms
32,004 KB |
testcase_17 | AC | 150 ms
30,988 KB |
testcase_18 | AC | 240 ms
77,136 KB |
testcase_19 | AC | 169 ms
39,568 KB |
testcase_20 | AC | 185 ms
33,780 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)