Submission #476171

#	Time	Username	Problem	Language	Result	Execution time	Memory
476171		quackson	Addk (eJOI21_addk)	C++11	0 / 100	135 ms	1928 KiB

Main

#include <bits/stdc++.h>
//#pragma GCC diagnostic warning "-std=c++11"
using namespace std;

int main(){
	long long n, k;
	cin >> n >> k;
	
	long long arr[n + 1];
	long long prefix[n + 1] = {0};
	long long prefix2[n + 1] = {0};
	for(int i = 1; i <= n; i++){
		cin >> arr[i];
		prefix[i] = prefix[i - 1] + arr[i];
		prefix2[i] = prefix2[i - 1] + arr[i] * i;
	}
	
	int q;
	cin >> q;
	while(q--){
		int oneOrTwo;
		cin >> oneOrTwo;
		
		if(oneOrTwo == 1){
			int smth;
			cin >> smth;
		}
		
		else{
			int l, r, m;
			cin >> l >> r >> m;
			
			m = min(r - l + 2 - m, m);
			
			int a = (prefix2[l + m - 1] - prefix2[l - 1]) - (prefix[l + m - 1] - prefix[l - 1]) * (l - 1);
			int b = (prefix[r - m] - prefix[l + m - 1]) * m;
			int c = (prefix[r] - prefix[r - m]) * (r + 1) - (prefix2[r] - prefix2[r - m]);
			
			cout << a + b + c << endl;
		}
	}
	return 0;
}

• Subtask 1: 0 / 36
• Subtask 2: 0 / 56
• Subtask 3: 0 / 8