結果
問題 | No.206 数の積集合を求めるクエリ |
ユーザー | kaoru murata |
提出日時 | 2021-08-28 12:05:17 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 1,848 ms / 7,000 ms |
コード長 | 1,560 bytes |
コンパイル時間 | 104 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 147,848 KB |
最終ジャッジ日時 | 2024-05-01 15:49:20 |
合計ジャッジ時間 | 55,485 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1,637 ms
126,168 KB |
testcase_01 | AC | 1,673 ms
125,940 KB |
testcase_02 | AC | 1,625 ms
126,048 KB |
testcase_03 | AC | 1,646 ms
126,188 KB |
testcase_04 | AC | 1,655 ms
126,316 KB |
testcase_05 | AC | 1,601 ms
126,164 KB |
testcase_06 | AC | 1,647 ms
126,296 KB |
testcase_07 | AC | 1,611 ms
126,004 KB |
testcase_08 | AC | 1,606 ms
126,296 KB |
testcase_09 | AC | 1,555 ms
126,312 KB |
testcase_10 | AC | 1,541 ms
126,188 KB |
testcase_11 | AC | 1,616 ms
126,044 KB |
testcase_12 | AC | 1,635 ms
126,048 KB |
testcase_13 | AC | 1,573 ms
126,300 KB |
testcase_14 | AC | 1,640 ms
126,652 KB |
testcase_15 | AC | 1,585 ms
126,316 KB |
testcase_16 | AC | 1,589 ms
126,304 KB |
testcase_17 | AC | 1,743 ms
147,208 KB |
testcase_18 | AC | 1,697 ms
142,804 KB |
testcase_19 | AC | 1,687 ms
146,920 KB |
testcase_20 | AC | 1,640 ms
142,296 KB |
testcase_21 | AC | 1,648 ms
143,572 KB |
testcase_22 | AC | 1,664 ms
144,564 KB |
testcase_23 | AC | 1,713 ms
147,848 KB |
testcase_24 | AC | 1,802 ms
147,216 KB |
testcase_25 | AC | 1,760 ms
146,848 KB |
testcase_26 | AC | 1,721 ms
142,476 KB |
testcase_27 | AC | 1,696 ms
141,060 KB |
testcase_28 | AC | 1,744 ms
144,660 KB |
testcase_29 | AC | 1,848 ms
144,248 KB |
testcase_30 | AC | 1,809 ms
141,028 KB |
ソースコード
import sys import numpy as np def read_int(): return int(input()) def read_int_tuple(): return map(int, sys.stdin.readline().rstrip().split()) def read_int_list(): return list(map(int, input().split())) def convolute(As, Bs, MOD=10 ** 9 + 7, fft_len=2 ** 20): """ As, Bs, np.array([..], dtype=np.int64) """ def _convolute8(aa, bb): return np.round(np.real(np.fft.irfft(np.fft.rfft(aa, fft_len) * np.fft.rfft(bb, fft_len), fft_len))).astype(np.int64) def _convolute16(aa, bb): mask = (1 << 8) - 1 uAs = aa >> 8 uBs = bb >> 8 dAs = aa & mask dBs = bb & mask uCs = _convolute8(uAs, uBs) % MOD dCs = _convolute8(dAs, dBs) % MOD mCs = (_convolute8((uAs + dAs), (uBs + dBs)) - uCs - dCs) % MOD return (uCs * pow(2, 8 * 2, MOD) + mCs * pow(2, 8, MOD) + dCs) % MOD mask = (1 << 16) - 1 uAs = As >> 16 uBs = Bs >> 16 dAs = As & mask dBs = Bs & mask uCs = _convolute16(uAs, uBs) % MOD dCs = _convolute16(dAs, dBs) % MOD mCs = (_convolute16((uAs + dAs), (uBs + dBs)) - uCs - dCs) % MOD return (uCs * pow(2, 16 * 2, MOD) + mCs * pow(2, 16, MOD) + dCs) % MOD L, M, N = read_int_tuple() A = read_int_list() B = read_int_list() Q = read_int() C = [0] * (N + 1) D = [0] * (N + 1) for a in A: C[a] = 1 for b in B: D[N - b] = 1 ans = convolute(np.array(C, dtype=np.int64), np.array(D, dtype=np.int64)) for i in range(Q): print(ans[N + i])