oj.uz

Submission #958794

# Submission time Handle Problem Language Result Execution time Memory
958794 2024-04-06T16:59:43 Z Acanikolic Addk (eJOI21_addk) C++14
92 / 100
 2000 ms 13700 KB 
#include <bits/stdc++.h>
			 				 
#define pb push_back 
			
#define F first
		 
#define S second
			 
#define int long long 			 
			 
using namespace std;
			 
const int N = 1e5 + 10;
			 
const int mod = 998244353;

int st[N * 4],lazy[N * 4];

void push(int node,int tl,int tr) {
	st[node] += (tr - tl + 1) * lazy[node];
	if(tl != tr) {
		lazy[node * 2] += lazy[node];
		lazy[node * 2 + 1] += lazy[node];
	}
	lazy[node] = 0;
}

void modify(int node,int tl,int tr,int l,int r,int val) {
	push(node,tl,tr);
	if(tl > r || tr < l) return;
	if(tl >= l && tr <= r) {
		lazy[node] += val;
		push(node,tl,tr);
		return;
	}
	push(node,tl,tr);
	int mid = (tl + tr) / 2;
	modify(node * 2,tl,mid,l,r,val);
	modify(node * 2 + 1,mid + 1,tr,l,r,val);
	st[node] = st[node * 2] + st[node * 2 + 1];
}

int get(int node,int tl,int tr,int l,int r) {
	push(node,tl,tr);
	if(tl > r || tr < l) return 0;
	if(tl >= l && tr <= r) return st[node];
	int mid = (tl + tr) / 2;
	return get(node * 2,tl,mid,l,r) + get(node * 2 + 1,mid + 1,tr,l,r);
}

signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);

	int n,k;
	cin >> n >> k;
	vector<int>a(n + 1);
	for(int i = 1; i <= n; i++) cin >> a[i];
	vector<int>pref(n + 1),Pref(n + 1);
	for(int i = 1; i <= n; i++) pref[i] = pref[i - 1] + a[i];
	for(int i = 1; i <= n; i++) Pref[i] = Pref[i - 1] + pref[i];
	for(int i = 1; i <= n; i++) modify(1,1,n,i,i,pref[i]);
	int q;
	cin >> q;
	while(q--) {
		int type;
		cin >> type;
		if(type == 1) {
			vector<int>b(k + 1),c(k + 1);
			for(int i = 1; i <= k; i++) cin >> b[i];
			for(int i = 1; i <= k - 1; i++) c[i] = b[i + 1];
			c[k] = b[1];
			vector<int>cpy = a;
			for(int i = 1; i <= k; i++) {
				cpy[b[i]] = a[c[i]];
				modify(1,1,n,c[i],n,-a[c[i]]);
				modify(1,1,n,b[i],n,a[c[i]]);	
			}
			a = cpy;
		}else {
			int l,r,m,res = 0;
			cin >> l >> r >> m;
			int L = l + m - 1;
			if(L <= r) res += get(1,1,n,1,r);
			if(L <= r) res -= get(1,1,n,1,L - 1);
			//if(L <= r) res += Pref[r] - Pref[L - 1];
			int R = r - m + 1;
			if(R >= l) res -= get(1,1,n,1,R - 1);
			//if(R >= l) res -= Pref[R - 1];
			//if(R >= l && l >= 2) res += Pref[l - 2];
			if(R >= l && l >= 2) res += get(1,1,n,1,l - 2);
			cout << res << "\n"; 
		}
	}
   	return 0;
}

Subtask #1
36.0 / 36.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2392 KB Output is correct
2 Correct 4 ms 2392 KB Output is correct
3 Correct 3 ms 2396 KB Output is correct
4 Correct 4 ms 2652 KB Output is correct
5 Correct 5 ms 2648 KB Output is correct
6 Correct 6 ms 2652 KB Output is correct
7 Correct 8 ms 2908 KB Output is correct
8 Correct 9 ms 2904 KB Output is correct
9 Correct 13 ms 4952 KB Output is correct

Subtask #2
56.0 / 56.0

# Verdict Execution time Memory Grader output
1 Correct 37 ms 5564 KB Output is correct
2 Correct 53 ms 5988 KB Output is correct
3 Correct 86 ms 6420 KB Output is correct
4 Correct 215 ms 8708 KB Output is correct
5 Correct 372 ms 10380 KB Output is correct
6 Correct 775 ms 10340 KB Output is correct
7 Correct 749 ms 9928 KB Output is correct

Subtask #3
0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 482 ms 6612 KB Output is correct
2 Correct 506 ms 11652 KB Output is correct
3 Execution timed out 2054 ms 13700 KB Time limit exceeded
4 Halted 0 ms 0 KB -