結果

問題 No.2568 列辞書順列列
ユーザー Tatsu_mr
提出日時 2024-09-18 00:41:45
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 273 ms / 3,000 ms
コード長 929 bytes
コンパイル時間 5,178 ms
コンパイル使用メモリ 310,804 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-18 00:42:01
合計ジャッジ時間 15,328 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;

using mint = atcoder::modint998244353;

template <class T>
struct prefixsum {
    int n;
    vector<T> s;
    
    prefixsum() {}
    prefixsum(vector<T> a) : n(a.size()), s(n + 1) {
        for (int i = 0; i < n; i++) {
            s[i + 1] = s[i] + a[i];
        }
    }
   
    //[l, r)
    T get(int l, int r) {
        return s[r] - s[l];
    }
};

int main() {
    int n, m, q;
    cin >> n >> m >> q;
    vector<int> a(n);
    for (int i = 0; i < n; i++) {
        cin >> a[i];
        a[i]--;
    }
    vector<mint> cnt(n);
    mint cur = 1;
    for (int i = n - 1; i >= 0; i--) {
        cnt[i] = mint(a[i]) * cur;
        cur *= mint(m);
    }
    prefixsum<mint> s(cnt);
    while (q--) {
        int l, r;
        cin >> l >> r;
        l--;
        mint ans = s.get(l, r) / mint(m).pow(n - r) + mint(1);
        cout << ans.val() << endl;
    }
}
0