#941127 (C++17) No.2571 列辞書順列

• 問題: No.2571 列辞書順列
• ユーザー: kwm_t
• 提出日時: 2023-12-30 02:38:58
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: RE
• 実行時間: -
• コード長: 1,882 bytes
• コンパイル時間: 5,347 ms
• コンパイル使用メモリ: 271,188 KB
• 実行使用メモリ: 8,444 KB
• 最終ジャッジ日時: 2023-12-30 02:39:14
• 合計ジャッジ時間: 14,231 ms
• ジャッジサーバーID （参考情報）: judge14 / judge13

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	5 ms; 6,676 KB
testcase_01	AC	5 ms; 6,676 KB
testcase_02	RE	-
testcase_03	RE	-
testcase_04	AC	80 ms; 6,676 KB
testcase_05	AC	77 ms; 6,676 KB
testcase_06	RE	-
testcase_07	RE	-
testcase_08	RE	-
testcase_09	RE	-
testcase_10	RE	-
testcase_11	RE	-
testcase_12	RE	-
testcase_13	RE	-
testcase_14	RE	-
testcase_15	RE	-
testcase_16	RE	-
testcase_17	RE	-
testcase_18	RE	-
testcase_19	RE	-
testcase_20	RE	-
testcase_21	RE	-
testcase_22	RE	-
testcase_23	RE	-
testcase_24	RE	-
testcase_25	RE	-
testcase_26	RE	-
testcase_27	RE	-
testcase_28	RE	-
testcase_29	RE	-
testcase_30	AC	5 ms; 6,676 KB

ソースコード

#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 * pw[k - (r - l) + 1];
			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;
}