結果
| 問題 | No.723 2つの数の和 |
| ユーザー |
 tktk_snsn
|
| 提出日時 | 2020-12-30 14:12:05 |
| 言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 478 bytes |
| コンパイル時間 | 367 ms |
| コンパイル使用メモリ | 12,544 KB |
| 実行使用メモリ | 86,596 KB |
| 最終ジャッジ日時 | 2024-04-16 12:32:26 |
| 合計ジャッジ時間 | 6,651 ms |
|
ジャッジサーバーID （参考情報） |
judge2 / judge5 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | TLE | - |
| testcase_01 | -- | - |
| testcase_02 | -- | - |
| testcase_03 | -- | - |
| testcase_04 | -- | - |
| testcase_05 | -- | - |
| testcase_06 | -- | - |
| testcase_07 | -- | - |
| testcase_08 | -- | - |
| testcase_09 | -- | - |
| testcase_10 | -- | - |
| testcase_11 | -- | - |
| testcase_12 | -- | - |
| testcase_13 | -- | - |
| testcase_14 | -- | - |
| testcase_15 | -- | - |
| testcase_16 | -- | - |
| testcase_17 | -- | - |
| testcase_18 | -- | - |
| testcase_19 | -- | - |
| testcase_20 | -- | - |
| testcase_21 | -- | - |
| testcase_22 | -- | - |
| testcase_23 | -- | - |
| testcase_24 | -- | - |
ソースコード
import numpy as np
from scipy.signal import fftconvolve
def convolve(A, B):
n = len(A) + len(B) + 1
fft_len = 1 << (n - 1).bit_length()
fa = np.fft.rfft(A, fft_len)
fb = np.fft.rfft(B, fft_len)
res = np.fft.irfft(fa * fb, fft_len)[:n + 1]
return np.rint(res + 0.01).astype(np.int32)
N, X = map(int, input().split())
A = np.array(input().split(), dtype=np.int32)
A = np.bincount(A)
F = convolve(A, A)
if X < len(F):
print(F[X])
else:
print(0)