結果
問題 | No.206 数の積集合を求めるクエリ |
ユーザー | kaoru murata |
提出日時 | 2021-08-28 12:05:17 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 2,001 ms / 7,000 ms |
コード長 | 1,560 bytes |
コンパイル時間 | 122 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 148,944 KB |
最終ジャッジ日時 | 2024-11-21 20:53:33 |
合計ジャッジ時間 | 63,232 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1,883 ms
126,240 KB |
testcase_01 | AC | 1,869 ms
126,532 KB |
testcase_02 | AC | 1,850 ms
126,304 KB |
testcase_03 | AC | 1,800 ms
126,292 KB |
testcase_04 | AC | 1,776 ms
126,344 KB |
testcase_05 | AC | 1,829 ms
126,596 KB |
testcase_06 | AC | 1,808 ms
126,276 KB |
testcase_07 | AC | 1,830 ms
127,188 KB |
testcase_08 | AC | 1,845 ms
126,256 KB |
testcase_09 | AC | 1,768 ms
127,712 KB |
testcase_10 | AC | 1,813 ms
127,636 KB |
testcase_11 | AC | 1,794 ms
126,836 KB |
testcase_12 | AC | 1,830 ms
127,108 KB |
testcase_13 | AC | 1,776 ms
127,016 KB |
testcase_14 | AC | 1,777 ms
127,184 KB |
testcase_15 | AC | 1,750 ms
126,704 KB |
testcase_16 | AC | 1,755 ms
126,772 KB |
testcase_17 | AC | 1,984 ms
148,944 KB |
testcase_18 | AC | 1,926 ms
142,576 KB |
testcase_19 | AC | 1,841 ms
148,184 KB |
testcase_20 | AC | 1,841 ms
142,488 KB |
testcase_21 | AC | 1,807 ms
143,952 KB |
testcase_22 | AC | 1,852 ms
144,944 KB |
testcase_23 | AC | 1,929 ms
148,836 KB |
testcase_24 | AC | 2,001 ms
147,824 KB |
testcase_25 | AC | 1,975 ms
147,760 KB |
testcase_26 | AC | 1,955 ms
143,776 KB |
testcase_27 | AC | 1,903 ms
143,136 KB |
testcase_28 | AC | 1,953 ms
145,168 KB |
testcase_29 | AC | 1,940 ms
143,908 KB |
testcase_30 | AC | 1,941 ms
142,464 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])