結果

問題 No.2369 Some Products
ユーザー gew1fw
提出日時 2025-06-12 18:56:31
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 941 bytes
コンパイル時間 275 ms
コンパイル使用メモリ 82,328 KB
実行使用メモリ 53,436 KB
最終ジャッジ日時 2025-06-12 18:56:38
合計ジャッジ時間 4,217 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 1 TLE * 1 -- * 12
権限があれば一括ダウンロードができます

ソースコード

diff #

MOD = 998244353

def main():
    import sys
    input = sys.stdin.read
    data = input().split()
    ptr = 0
    n = int(data[ptr])
    ptr += 1
    p = list(map(int, data[ptr:ptr + n]))
    ptr += n
    p_mod = [x % MOD for x in p]
    q = int(data[ptr])
    ptr += 1
    queries = []
    for _ in range(q):
        a = int(data[ptr])
        b = int(data[ptr + 1])
        k = int(data[ptr + 2])
        queries.append((a, b, k))
        ptr += 3
    
    results = []
    for a, b, k in queries:
        a -= 1  # Convert to 0-based index
        elements = p_mod[a:b]
        m = b - a
        if k > m:
            results.append(0)
            continue
        dp = [0] * (k + 1)
        dp[0] = 1
        for x in elements:
            for i in range(k, 0, -1):
                dp[i] = (dp[i] + dp[i-1] * x) % MOD
        results.append(dp[k] % MOD)
    
    print('\n'.join(map(str, results)))

if __name__ == "__main__":
    main()
0