結果

問題 No.2369 Some Products
ユーザー gew1fw
提出日時 2025-06-12 14:36:14
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 890 bytes
コンパイル時間 178 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 52,884 KB
最終ジャッジ日時 2025-06-12 14:36:28
合計ジャッジ時間 4,309 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
    q = int(data[ptr])
    ptr += 1
    results = []
    for _ in range(q):
        a = int(data[ptr])
        b = int(data[ptr + 1])
        k = int(data[ptr + 2])
        ptr += 3
        a -= 1  # convert to 0-based
        sub = p[a:b]
        m = len(sub)
        if k > m or k == 0:
            results.append(0)
            continue
        dp = [0] * (k + 1)
        dp[0] = 1
        for x in sub:
            # Iterate from k down to 1
            for j in range(min(k, len(dp)-1), 0, -1):
                dp[j] = (dp[j] + x * dp[j-1]) % MOD
        results.append(dp[k] % MOD)
    
    sys.stdout.write('\n'.join(map(str, results)) + '\n')

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