結果

問題 No.2770 Coupon Optimization
ユーザー titiatitia
提出日時 2024-06-03 01:09:02
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 2,383 ms / 3,000 ms
コード長 416 bytes
コンパイル時間 545 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 122,864 KB
最終ジャッジ日時 2024-06-03 01:09:44
合計ジャッジ時間 36,169 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 841 ms
116,116 KB
testcase_01 AC 830 ms
116,124 KB
testcase_02 AC 845 ms
114,044 KB
testcase_03 AC 2,181 ms
122,812 KB
testcase_04 AC 2,166 ms
122,700 KB
testcase_05 AC 919 ms
115,076 KB
testcase_06 AC 2,210 ms
122,696 KB
testcase_07 AC 2,172 ms
119,956 KB
testcase_08 AC 2,251 ms
122,864 KB
testcase_09 AC 1,132 ms
115,296 KB
testcase_10 AC 1,607 ms
117,168 KB
testcase_11 AC 1,395 ms
119,360 KB
testcase_12 AC 1,670 ms
119,052 KB
testcase_13 AC 1,576 ms
116,816 KB
testcase_14 AC 2,262 ms
122,724 KB
testcase_15 AC 2,310 ms
122,772 KB
testcase_16 AC 2,383 ms
122,500 KB
testcase_17 AC 2,324 ms
122,468 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline

N,M=map(int,input().split())
A=list(map(int,input().split()))
B=list(map(int,input().split()))

for i in range(M):
    B[i]=100-B[i]

A.sort()
B.sort()

while len(B)<N:
    B.append(100)

import numpy
k=1<<19
g=numpy.fft.fft(A,k)
h=numpy.fft.fft(B,k)

f=[g[i]*h[i] for i in range(k)]

ANS=numpy.fft.ifft(f)

for ans in ANS[:N]:
    print(int(numpy.round(numpy.real(ans/100))))
0