結果

問題 No.206 数の積集合を求めるクエリ
ユーザー kaoru muratakaoru 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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])
0