# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
777114 | ind1v | Addk (eJOI21_addk) | C++11 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
const int N = 100005;
const int K = 15;
struct fenwick_tree {
int64_t fenw[N];
fenwick_tree() {
memset(fenw, 0, sizeof(fenw));
}
void upd(int idx, long long val) {
for (; idx < N; idx |= (idx + 1)) {
fenw[idx] += val;
}
}
int64_t get(int idx) {
int64_t res = 0;
for (; idx >= 0; idx &= (idx + 1), --idx) {
res += fenw[idx];
}
return res;
}
int64_t get(int l, int r) {
return get(r) - get(l - 1);
}
};
int n, k;
int a[N];
int64_t p[N];
int idx[K];
int q;
fenwick_tree ft1, ft2;
void upd(int idx, int64_t val) {
ft1.upd(idx, val);
ft2.upd(idx, 1LL * idx * val);
}
int64_t get(int idx) {
return 1LL * (idx + 1) * ft1.get(idx) - ft2.get(idx);
}
int64_t get(int l, int r) {
return get(r) - get(l - 1);
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
cin >> n >> k;
for (int i = 1; i <= n; i++) {
cin >> a[i];
p[i] = a[i] + p[i - 1];
}
for (int i = 1; i <= n; i++) {
upd(i, p[i] - p[i - 1]);
}
cin >> q;
while (q--) {
int t;
cin >> t;
if (t == 1) {
for (int i = 1; i <= k; i++) {
cin >> idx[i];
}
idx[k + 1] = idx[1];
for (int i = 1; i <= k; i++) {
upd(idx[i], a[idx[i + 1]] - a[idx[i]]);
}
int tmp = a[idx[1]];
for (int i = 1; i <= k - 1; i++) {
a[idx[i]] = a[idx[i + 1]];
}
a[idx[k]] = tmp;
} else if (t == 2) {
int l, r, m;
cin >> l >> r >> m;
cout << ft.get(min(l + m - 1, r), r) - ft.get(min(l + m - 1, r) - m, r - m) << '\n';
}
}
return 0;
}