結果

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

ソースコード

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