結果

問題 No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
ユーザー Salmonize
提出日時 2020-05-30 00:36:57
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 215 ms / 2,000 ms
コード長 879 bytes
コンパイル時間 100 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 13,000 KB
最終ジャッジ日時 2024-11-06 10:03:05
合計ジャッジ時間 3,503 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys, heapq as hq

readline = sys.stdin.readline

k = 70
K = 1<<k

ns = lambda: readline().rstrip()
ni = lambda: int(readline().rstrip())
nm = lambda: map(int, readline().split())
nl = lambda: list(map(int, readline().split()))
nu = lambda L: int("".join([bin(K+a)[-k:] for a in L[::-1]]), 2)
st = lambda n: bin(n)[2:] + "0"
li = lambda s, p, q: [int(a, 2) if len(a) else 0 for a in [s[-(i+1)*k-1:-i*k-1] for i in range(p+q-1)]]


def solve():
    mod = 998244353
    n, q = nm()
    a = nl()
    def recurr(l, r):
        if l + 1 == r:
            return [a[l]-1, 1]
        res1 = recurr(l, (l+r)//2)
        res2 = recurr((l+r)//2, r)
        res = li(st(nu(res1) * nu(res2)), len(res1), len(res2))
        for i in range(len(res)):
            res[i] %= mod
        return res
    p = recurr(0, n)
    b = nl()
    for x in b:
        print(p[x])

    return

solve()
0