Submission #474724

#	Time	Username	Problem	Language	Result	Execution time	Memory
474724		fuad27	Addk (eJOI21_addk)	C++17	92 / 100	314 ms	6580 KiB

Main

#include<bits/stdc++.h>
#define int long long
using namespace std;
int32_t main () {
	int n, k;
	cin >> n >> k;
	int arr[n+1], prefix1[n+1] = {0}, prefix2[n+1] = {0};
	for(int i = 1;i<=n;i++) {
		cin >> arr[i];
		prefix1[i] = prefix1[i-1] + arr[i];
		prefix2[i] = prefix2[i-1] + arr[i]*i;
	}
	int q;
	cin >> q;
	while(q--) {
		int k;
		cin >> k;
		if(k == 1) {
			int d;
			cin >> d;
		}
		else {
			int ans = 0;
			int l, r, m;
			cin >> l >> r >> m;
			m = min(r-l-m + 2, m);
			int a = (prefix2[l+m-1] - prefix2[l-1])-(prefix1[l+m-1] - prefix1[l-1])*(l-1);
			int b = (prefix1[r-m] - prefix1[l+m-1])*m;
			int c = (prefix1[r] - prefix1[r-m])*(r+1) - (prefix2[r] - prefix2[r-m]);
		       cout<<a+b+c<<endl;
		}
	}
}

Compilation message (stderr)

Main.cpp: In function 'int32_t main()':
Main.cpp:23:8: warning: unused variable 'ans' [-Wunused-variable]
   23 |    int ans = 0;
      |        ^~~

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