結果

問題 No.2568 列辞書順列列
ユーザー けんぴん
提出日時 2023-10-26 02:34:38
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 408 ms / 3,000 ms
コード長 775 bytes
コンパイル時間 1,013 ms
コンパイル使用メモリ 82,444 KB
実行使用メモリ 132,904 KB
最終ジャッジ日時 2024-09-25 16:07:33
合計ジャッジ時間 12,793 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

def main():
    mod = 998244353
    N,M,Q = map(int,input().split())
    assert 1 <= N <= 2*10**5
    assert 1 <= M <= 2*10**5
    assert 1 <= Q <= 2*10**5
    A = list(map(lambda x: int(x)-1, input().split()))
    for a in A:
        assert 0 <= a <= M-1
    acc_A = [0]
    e = 1
    for a in A[::-1]:
        acc_A.append((acc_A[-1] + a*e) % mod)
        e = (e * M) % mod
    # Mの逆元
    inv_M = pow(M, mod-2, mod)
    # Mのi乗の逆元を前計算
    inv_M_list = [1]
    for i in range(1, N+1):
        inv_M_list.append((inv_M_list[-1] * inv_M) % mod)
    for _ in range(Q):
        l,r = map(int, input().split())
        assert 1 <= l <= r <= N
        print(((acc_A[-l] - acc_A[-r-1]) * inv_M_list[N-r] + 1) % mod)
    
if __name__ == "__main__":
    main()
0