oj.uz

Submission #766039

# Submission time Handle Problem Language Result Execution time Memory
766039 2023-06-25T09:13:47 Z vjudge1 Addk (eJOI21_addk) C++14
92 / 100
 38 ms 6764 KB 
//srand(time(0)) - always changing
//order_of_key(k): Number of items strictly smaller than k .
//find_by_order(k): K-th element in a set (counting from zero).
//#pragma GCC optimize("O3,unroll-loops")
//#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>

#define ordered_set tree<ll,null_type,less<ll>,rb_tree_tag,tree_order_statistics_node_update>
#define fopen(s) freopen(s".in","r",stdin);freopen(s".out","w",stdout)
#define all(x) (x).begin(), (x).end()
#define len(x) (int)x.size()
#define pb push_back
#define pf push_front
#define popb pop_back
#define popf pop_front
#define elif else if
#define F first
#define S second
#define int long long

typedef unsigned long long ull;
typedef long long ll;

using namespace std;
using namespace __gnu_pbds;

const int MOD = 1e9 + 7;
const int N = 2e5 + 7;
const int P = 911;
const ll INF = 1e18;

int gcd(int a, int b) {
	while (b) {
		a %= b;
		swap (a, b);
	}
	return a;
}

ll __sqrt(ll x) {
	ll result = 0;
	for (ll k = 1ll << 30; k != 0; k >>= 1) {
		if ((result + k) * (result + k) <= x) {
			result += k;
		}
	}
	return result;
}

int a[N], ps[N], ss[N], sum[N];

void solve() {
	int n, k;
	cin >> n >> k;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		ps[i] = ps[i - 1] + a[i] * i;
		sum[i] = sum[i - 1] + a[i];
	}
	for (int i = n; i > 0; i--) {
		ss[i] = ss[i + 1] + a[i] * (n - i + 1);
	}
	int q;
	cin >> q;
	while (q--) {
		int ty, l, r, m;
		cin >> ty >> l;
		if (ty == 1) continue;
		cin >> r >> m;
		int ln = r - l + 1;
		if (ln < m * 2 - 1) {
			int act = ln - m + 1;
			int lmx = l + act - 2;
			int rmx = r - act + 2;
			int ans1 = (ps[lmx] - ps[l - 1]) - (sum[lmx] - sum[l - 1]) * (l - 1);
			int ans2 = 0;
			if (lmx + 1 <= rmx - 1) {
				ans2 = (sum[rmx - 1] - sum[lmx]) * act;
			}
			int ans3 = (ss[rmx] - ss[r + 1]) - (sum[r] - sum[rmx - 1]) * (n - r);
			cout << ans1 + ans2 + ans3 << '\n';
//			cout << ans1 << ' ' << ans2 << ' ' << ans3 << '\n';
		} else {
			int ans1 = 0;
			if (m > 1) ans1 = (ps[l + m - 2] - ps[l - 1]) - (sum[l + m - 2] - sum[l - 1]) * (l - 1);
			int ans2 = (sum[r] - sum[l - 1]) * m;
			if (m > 1) ans2 = (sum[r - m + 1] - sum[l + m - 2]) * m;
			int ans3 = 0;
			if (m > 1) ans3 = (ss[r - m + 2] - ss[r + 1]) - (sum[r] - sum[r - m + 1]) * (n - r);
			cout << ans1 + ans2 + ans3 << '\n';
//			cout << ans1 << ' ' << ans2 << ' ' << ans3 << '\n';
//			cout << "ans3 =  " << ss[r - m + 2] << ' ' << ss[r + 1] << ' ' << sum[r] << ' ' << sum[r - m + 1] << ' ' << n - r << '\n';
		}
	}
}

const bool Cases = 0;

signed main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int TT = 1;
	if (Cases) cin >> TT;
	while (TT--) {
		solve();
	}
}

Subtask #1
36.0 / 36.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 340 KB Output is correct
2 Correct 1 ms 340 KB Output is correct
3 Correct 1 ms 468 KB Output is correct
4 Correct 2 ms 468 KB Output is correct
5 Correct 2 ms 592 KB Output is correct
6 Correct 2 ms 596 KB Output is correct
7 Correct 3 ms 724 KB Output is correct
8 Correct 3 ms 724 KB Output is correct
9 Correct 4 ms 980 KB Output is correct

Subtask #2
56.0 / 56.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 1224 KB Output is correct
2 Correct 11 ms 2396 KB Output is correct
3 Correct 16 ms 3128 KB Output is correct
4 Correct 26 ms 5072 KB Output is correct
5 Correct 38 ms 6764 KB Output is correct
6 Correct 36 ms 6544 KB Output is correct
7 Correct 38 ms 6580 KB Output is correct

Subtask #3
0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 20 ms 2644 KB Output isn't correct
2 Halted 0 ms 0 KB -