結果
問題 | No.2571 列辞書順列 |
ユーザー |
![]() |
提出日時 | 2023-12-30 02:42:07 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 124 ms / 3,000 ms |
コード長 | 1,895 bytes |
コンパイル時間 | 4,161 ms |
コンパイル使用メモリ | 257,664 KB |
最終ジャッジ日時 | 2025-02-18 15:25:24 |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 29 |
ソースコード
#include <bits/stdc++.h>#include <atcoder/all>using namespace std;using namespace atcoder;//using mint = modint1000000007;//const int mod = 1000000007;using mint = modint998244353;const int mod = 998244353;//const int INF = 1e9;//const long long LINF = 1e18;#define rep(i, n) for (int i = 0; i < (n); ++i)#define rep2(i,l,r)for(int i=(l);i<(r);++i)#define rrep(i, n) for (int i = (n-1); i >= 0; --i)#define rrep2(i,l,r)for(int i=(r-1);i>=(l);--i)#define all(x) (x).begin(),(x).end()#define allR(x) (x).rbegin(),(x).rend()#define endl "\n"#define P pair<int,int>template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; }template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; }vector<mint>pw(300005);struct S {mint val;int sz;S(mint val, int sz) :val(val), sz(sz) {}};S op(S a, S b) {mint val = a.val * pw[b.sz] + b.val;int sz = a.sz + b.sz;return S(val, sz);}S e() { return S(0, 0); }mint op2(mint a, mint b) { return a + b; }mint e2() { return 0; }int main() {ios::sync_with_stdio(false);cin.tie(nullptr);int n, m, k, q; cin >> n >> m >> k >> q;pw[0] = 1;rep2(i, 1, pw.size()) pw[i] = m * pw[i - 1];vector<int>a(n);rep(i, n)cin >> a[i], a[i]--;segtree<S, op, e>seg(n);rep(i, n) seg.set(i, { a[i],1 });segtree<mint, op2, e2>seg2(n);rep(i, n) seg2.set(i, a[i]);while (q--) {int t; cin >> t;if (1 == t) {int l, r; cin >> l >> r;l--;if (1 == m) {cout << r - l << endl;continue;}mint ans = 0;ans += seg.prod(l, r).val * pow_mod(m, k - (r - l) + 1, mod);ans -= seg2.prod(l, r);ans /= (m - 1);ans += r - l;cout << ans.val() << endl;}else {int x, y; cin >> x >> y;x--, y--;seg.set(x, { y,1 });seg2.set(x, y);}}return 0;}